剑指offer-62-二叉搜索树的第k个结点

题目

给定一棵二叉搜索树，请找出其中的第k小的结点。例如， （5，3，7，2，4，6，8） 中，按结点数值大小顺序第三小结点的值为4。

思路

二叉搜索树的中序遍历是递增序列，利用栈（先入后出）中序遍历，当count==k返回即可

代码

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
import java.util.Stack;
public class Solution {
    //二叉搜索树的中序遍历是递增序列
    //利用栈中序遍历
    TreeNode KthNode(TreeNode pRoot, int k)
    {
         if(pRoot == null || k==0){
            return null;
        }
        int count = 0;
        Stack<TreeNode> stack = new Stack<>();
        TreeNode temp = pRoot;
        while(!stack.isEmpty()||temp != null){
            count++;
            while(temp != null){//将所有的左子树放入栈
                stack.push(temp);
                temp = temp.left;
            }
            temp = stack.pop();//取出栈顶
            if(count == k){
                return temp;
            }
            temp = temp.right;//右子树
        }
        return null;
    }


}