二叉搜索树的第 K 个结点

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

分析：二叉排序树，中序遍历是递增序列

PS：与牛客相同！

代码：

import java.util.Stack;

public class Solution{
    TreeNode KthNode(TreeNode root, int k){
        if(root == null || k == 0){
            return null;
        }
        Stack<TreeNode> stack = new Stack<>();
        int count = 0;
        TreeNode cur = root;
        while(!stack.isEmpty() || cur != null){
            while(cur != null){
                stack.push(cur);
                cur = cur.left;
            }
            cur = stack.pop();
            count++;    
            if(count == k){
                return cur;
            }
            cur = cur.right;
        }
        return null;
    }
}