*LeetCode-Kth Smallest Element in a BST

来源:互联网 发布:数学符号输入软件 编辑:程序博客网 时间:2024/06/05 09:29

一个函数来数这个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 );    }}


0 0
原创粉丝点击