题目描述
给定一颗二叉搜索树,请找出其中的第k小的结点。例如, 5 / \ 3 7 /\ /\ 2 4 6 8 中,按结点数值大小顺序第三个结点的值为4。
思路:
首先这个是一颗二叉搜索树,我只需要是的这个这颗二叉搜索树进行排序,然后找到第k个就可以了,那个我们都知道二叉搜索树的中序遍历是一个有顺序的序列,这个时候我就是要中序遍历这颗二叉树,然后设置一个变量,访问一个变量的时候就加一,判断这个变量和k是否相等,如果相等,则将当前的这个结点返回即可。
实现:
import java.util.Stack;
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
TreeNode KthNode(TreeNode pRoot, int k)
{
//首先中序遍历这颗二叉树
Stack<TreeNode> stack = new Stack<TreeNode>();
TreeNode p = pRoot;
int i = 0;
while(p != null || !stack.isEmpty()){
if(p != null){
stack.push(p);
p = p.left;
}else{
TreeNode node = stack.pop();
i++;
if(i == k){
return node;
}
p = node.right;
}
}
return null;
}
}