剑指 二叉搜索树的第k个节点

题目描述

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

运行时间：28ms

占用内存：9696k

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

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

    }

}
*/
public class Solution {
    //二叉搜索树的中序遍历即为从大到小的顺序
    int count = 0; //计数器存储已经访问过的节点的个数
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot == null)
            return null;
        TreeNode node = KthNode(pRoot.left, k);
        if(node != null)//左子树中找到了目标节点
            return node;
        count++; //访问了当前pRoot节点 计数器+1
        if (count == k)
            return pRoot; //计数器达到需要的序数值后，返回当前节点；
        node = KthNode(pRoot.right, k);
        if(node != null)
            return node;
        return null;
    }


}