二叉搜索树的第k个结点
题目描述
给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8) 中,按结点数值大小顺序第三小结点的值为4。 ***
思路
当然可以把二叉搜索树每个点都遍历一遍,然后存在数组中,再排个序,取出第k个。 但熟悉概念的话可以知道,二叉搜索树的中序遍历就是从小到大遍历,所以当中序遍历到第k个时,即可返回。
Java实现
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
private int index = 0;
private TreeNode ans = null;
TreeNode KthNode(TreeNode pRoot, int k)
{
if(ans != null){
return ans;
}
if(pRoot==null || k <= 0){
return null;
}
if(pRoot.left!=null){
ans = KthNode(pRoot.left, k);
}
index++;
if(index == k){
ans = pRoot;
return ans;
}
if(pRoot.right!=null){
ans = KthNode(pRoot.right, k);
}
return ans;
}
}