描述
给定一棵结点数为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的结点即可。
示例1
输入:
{5,3,7,2,4,6,8},3
复制返回值:
4
复制
示例2
输入:
{},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;
}
预备工作:创建java形式的BST,和C++稍有区别
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<root.val) insert(root.left,val);
else insert(root.right,val);*/
//注意直接这么写 不行 这不像c++的 node* &root 直接是取个别名(地址(就是引用)的别名 两套机制★★★) java没有这套机制
//当root.left不是null时有效 当他等于null时 就连不起来了 ★★★
if(val<root.val){
if(root.left==null) root.left=new TreeNode(val);
else insert(root.left,val);
}else{
if(root.right==null) root.right=new TreeNode(val);
else insert(root.right,val);
}
}
public static TreeNode createBSTree(int[] vals){
TreeNode root=new TreeNode(vals[0]);//第一个就是根 直接插入
for (int i = 1; i < vals.length; i++) {
insert(root,vals[i]);//java此处不允许传入一个root=null
}
return root;
}
///
public void preorder(){
System.out.print(this.val+" ");
if(this.left!=null) this.left.preorder();
if(this.right!=null) this.right.preorder();
}
public void Inorder(){
if(this.left!=null) this.left.Inorder();
System.out.print(this.val+" ");
if(this.right!=null) this.right.Inorder();
}
public void levelOrder(){
Queue<TreeNode> q=new LinkedList<TreeNode>();//注意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);
}
}
}