您的位置:首页 > 其它

leetcode_middle_26_230. Kth Smallest Element in a BST

2017-02-07 16:12 344 查看
题意:

给一个二叉排序树(bst),找出其中第k小的结点

分析:

我们就要去找从小到大是什么样的顺序,发现正好就是从上到下从左到右回溯搜索的顺序,那么递归深搜即可。那么我们对结点计数是在哪里呢?

我们来看遍历的过程中:到达当前结点,向左搜索,返回,向右搜索,返回。我们发现当前结点被走了三次,但是,并不意味着这三个位置计数都可以。虽然次数都只是一次,但是计数的顺序有区别。我们对一个结点的计数应该在搜索左边之后,搜索右边之前。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    int k = 0;
    int val = 0;
    public int kthSmallest(TreeNode root, int k) {
        this.k = k;
        helper(root);
        return val;
    }
    
    private void helper(TreeNode node){
        if(node == null){  //叶子节点的下一个结点(不满足条件的结点)
            return;
        }
                             //到达当前结点
        helper(node.left);   //向下(左)搜素
                             //从下(左)返回当前结点
        k--;
        if(k == 0){
            val = node.val;
        }
        helper(node.right);  //向下(右)搜索
                             //从下(右)返回当前结点
    }
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  leetcode