程序员人生 网站导航

二叉搜索树的第k个结点

栏目:php教程时间:2016-08-22 09:15:40

题目

给定1颗2叉搜索树,请找出其中的第k大的结点。

解题

中序遍用时候找到第k大结点

import java.util.ArrayList; public class Solution { ArrayList<TreeNode> list = new ArrayList<TreeNode>(); TreeNode KthNode(TreeNode pRoot, int k) { inorder(pRoot); if(k<=0 || k> list.size()) return null; return list.get(k-1); } public void inorder(TreeNode root){ if(root == null) return; inorder(root.left); list.add(root); inorder(root.right); } }

利用中序遍历,记录遍历结点个数找到第k个结点

/* public class TreeNode { int val = 0; TreeNode left = null; TreeNode right = null; public TreeNode(int val) { this.val = val; } } */ public class Solution { int count = 0; // 记录遍历结点个数 TreeNode KthNode(TreeNode root, int k) { if(root==null|| k<=0) return null; TreeNode left = KthNode(root.left,k); if(left!=null) return left; count++; if(count == k) return root; TreeNode right = KthNode(root.right,k); if(right!=null) return right; return null; } }
------分隔线----------------------------
------分隔线----------------------------

最新技术推荐