描述
给定一棵结点数为n 二叉搜索树,请找出其中的第 k 小的TreeNode结点值。
1.返回第k小的节点值即可
2.不能查找的情况,如二叉树为空,则返回-1,或者k大于n等等,也返回-1
3.保证n个节点的值不一样
数据范围: 0≤n≤10000 \le n \le10000≤n≤1000,0≤k≤10000 \le k \le10000≤k≤1000,树上每个结点的值满足0≤val≤10000 \le val \le 10000≤val≤1000
进阶:空间复杂度 O(n)O(n)O(n),时间复杂度 O(n)O(n)O(n)
如输入{5,3,7,2,4,6,8},3时,二叉树{5,3,7,2,4,6,8}如下图所示:
该二叉树所有节点按结点值升序排列后可得[2,3,4,5,6,7,8],所以第3个结点的结点值为4,故返回对应结点值为4的结点即可。
这道题最直观的解法就是把二叉树里的节点按从大到小排续,然后找第k个,由于二叉树的中序遍历就可以得出升序数组,所以就不用遍历整个二叉树并排序了,只需要进行中序遍历,遍历了一个节点就让k-1,这里我选择用提升作用域的方法,定义了一个全局变量n,让他记住k的值,然后在中序遍历二叉树的函数中判断n是否为0,当他是0的时候当前节点就是我们要找的节点,同时为了记住找到的节点值,我会把中序遍历的左右子节点返回的结果相加,如果大于-2,就说明找到了节点,因为没找到就会返回-1,左右都没找到就会返回-2,找到了第k个节点的时候他的返回值就是第k个节点的val值加上-1,是大于-2的
Java代码
private int n;
public int KthNode (TreeNode proot, int k) {
n=k;
return SNode(proot);
}
public int SNode(TreeNode proot){
if(n<=0||proot==null)return -1;
int result=0;
result+=SNode(proot.left);
n--;
if(n==1)return proot.val;
result+=SNode(proot.right);
if(result>-2)return result+1;
else return -1;
}