*LeetCode-Kth Smallest Element in a BST

bsbcarter

于 2015-09-27 07:53:41 发布

阅读量237

点赞数

分类专栏： tree leetcode DFS

本文链接：https://blog.csdn.net/bsbcarter/article/details/48764793

版权

leetcode 同时被 3 个专栏收录

220 篇文章 0 订阅

订阅专栏

tree

35 篇文章 0 订阅

订阅专栏

DFS

22 篇文章 0 订阅

订阅专栏

一个函数来数这个root下一共多少node 非常简单 recursive

然后就是主函数中判断left有多少node的时候需要数右边想清楚＋1 －1的事情因为要算上root本身

public class Solution {
    public int kthSmallest(TreeNode root, int k) {
        if ( countNode ( root.left ) < k - 1)
            return kthSmallest( root.right, k - 1 - countNode( root.left) );
        else if ( countNode ( root.left ) >= k)
            return kthSmallest( root.left, k);
        else
            return root.val;
    }
    public int countNode ( TreeNode node ){
        if ( node == null )
            return 0;
        return 1 + countNode ( node.left ) + countNode ( node.right );
    }
}