给定一棵结点数为n 二叉搜索树,请找出其中的第 k 小的TreeNode结点值。
1.返回第k小的节点值即可
2.不能查找的情况,如二叉树为空,则返回-1,或者k大于n等等,也返回-1
3.保证n个节点的值不一样
数据范围: 0≤n≤1000, 0≤k≤1000,树上每个结点的值满足 0≤val≤1000
进阶:空间复杂度 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的结点即可。
输入:
{5,3,7,2,4,6,8},3
复制返回值:
4
复制
输入:
{},1
返回值:
-1
当树是空
int n=0,ans=-1;
public void Inorder(TreeNode root,int k){
if(root==null||n>=k) return;//顺带剪枝
Inorder(root.left,k);
if((++n)==k) {//中序遍历到第四个即可
ans=root.val;
return;
}
Inorder(root.right,k);
}
public int KthNode (TreeNode proot, int k) {
Inorder(proot,k);
return ans;
}
c++的 node* &root; 是双重保险,指针的引用(指针的别名) 哪怕指针是是null也能连起来
java没有&引用 只有指针传递(虽然java里乘坐引用) 写BST的insert时不能照搬c++那一套,一旦root.left==null时,就完了,断链了
package cn.whu.jz.tree;
import java.util.LinkedList;
import java.util.Queue;
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(){}
public TreeNode(int val) {
this.val = val;
}
//作为二叉树插入 中序查找 失败节点插入
public static void insert(TreeNode root,int val){
if(root==null) {//查找失败节点就是插入结点
root=new TreeNode(val);
return;
}
/*if(val q=new LinkedList();//注意java队列 就是LinkedList
q.add(this);
while (!q.isEmpty()){//注意API
TreeNode p=q.poll();//pop就是poll
System.out.print(p.val+" ");//push就是add
if(p.left!=null) q.add(p.left);
if(p.right!=null) q.add(p.right);
}
}
}