描述:
给定一棵结点数为n 二叉搜索树,请找出其中的第 k 小的TreeNode结点值。
1.返回第k小的节点值即可
2.不能查找的情况,如二叉树为空,则返回-1,或者k大于n等等,也返回-1
3.保证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个位置的节点。
代码实现:
public class Solution {
public int KthNode (TreeNode proot, int k) {
if (proot == null) return -1;
List<Integer> list = new ArrayList<>();
MidSearch(proot,list);
if (k > list.size()) return -1;
for (int i=0;i<=list.size();++i){
if (i == k-1) {
return list.get(i);
}
}
return -1;
}
//中序遍历,得到有序序列
public void MidSearch(TreeNode proot, List<Integer> list){
if (proot == null) return;
MidSearch(proot.left,list);
list.add(proot.val);
MidSearch(proot.right,list);
}
}
思考:
网站上标的难度为中等难度,但其实着重考察的是对二叉搜索树,这种数据结构特性掌握的情况,其中用到了中序遍历,其他的像先序遍历、后续遍历都是需要掌握的,对应于不同的情况。