230. 二叉搜索树中第K小的元素

leetcode刷题

给定一个二叉搜索树,编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。
说明:
你可以假设 k 总是有效的,1 ≤ k ≤ 二叉搜索树元素个数。

class Solution {
    int count = 0;
    int res = -1;
    public int kthSmallest(TreeNode root, int k) {
        inOrder(root,k);
        return res;       
    }
    private void inOrder(TreeNode node,int k){
        if(node!=null){
            inOrder(node.left,k);
            count++;
            if(k==count){
                res = node.val;
                return;
            }
            inOrder(node.right,k);
        }
    }
}

错误写法:注意返回值是什么类型!!!

class Solution {
    public int kthSmallest(TreeNode node, int k) {
        if(node!=null){
            kthSmallest(node.left,k);//这样写为什么不对呢?是因为它的返回值就是int类型的了,而不是中序遍历了!!!1
            count++;
            if(k==count){
                res = node.val;
                return;
            }
            kthSmallest(node.right,k);
        }    
    }
}

你可能感兴趣的:(算法,二叉排序树)