NC81 二叉搜索树的第k个结点

描述

给定一棵二叉搜索树，请找出其中的第k小的TreeNode结点。

示例1

输入：

{5,3,7,2,4,6,8},3

复制返回值：

4

复制说明：

按结点数值大小顺序第三小结点的值为4   

非递归实现：二叉搜索树的第k个结点 中序遍历

import java.util.Stack;
public class Solution {
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot == null || k <= 0){
            return null;
        }
        Stack<TreeNode> stack = new Stack<>(); //建立栈
        TreeNode cur = pRoot;
        //while 部分为中序遍历
        while(!stack.isEmpty() || cur != null){ 
            if(cur != null){
                stack.push(cur); //当前节点不为null，应该寻找左儿子
                cur = cur.left;
            }else{
                cur = stack.pop();//当前节点null则弹出栈内元素，相当于按顺序输出最小值。
                if(--k == 0){ //计数器功能
                    return cur;
                }
                cur = cur.right;
            }
        }
        return null;
    }
}