题目描述
给定一颗二叉搜索树,请找出其中的第K小的节点。例如,(5,3,7,2,46,8)中,按节点数值大小顺序第三小节点值为4
解题思路分析
其实要解这道题首先需要了解什么是二叉搜索树,如果不知道二叉搜索树的相关性质的话,那解答这道题就很难了。话说回来,二叉搜索树其实就是一颗二叉树拥有这样的特点:所有父节点都比其左子节点大,比其右子节点小。知道这个性质之后,这道题马上就可以想出来了,只要中序遍历一遍这棵树便可以得到一个排好序的序列,然后找第k小的节点就很简单了。
代码实现
public static TreeNode KthNode(TreeNode pRoot, int k) {
if (pRoot == null || k <= 0) {
return null;
}
ArrayList<TreeNode> list = new ArrayList<TreeNode>();
// 中序遍历得到该二叉搜索树的中序遍历序列
inOrder(pRoot, list);
if (list.size() < k) {
return null;
}
return list.get(k - 1);
}
private static void inOrder(TreeNode pRoot, ArrayList<TreeNode> list) {
if (pRoot == null) {
return;
}
inOrder(pRoot.left, list);
list.add(pRoot);
inOrder(pRoot.right, list);
}