二叉搜索树的第K个节点（思路与实现）

题目描述

给定一颗二叉搜索树，请找出其中的第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;
    }
}