leetcode230. 二叉搜索树中第K小的元素(Kth Smallest Element in a BST)

给定一个二叉搜索树,编写一个函数kthSmallest来查找其中第k个最小的元素。

对于二叉搜索树,前序 中序遍历就是一个有序的搜索,因此只需要在前序 中序遍历中记录一下搜索次数就行了,比较容易实现

public static int num = 0;
    public static int remain = 0;
    public int kthSmallest(TreeNode root, int k) {
        remain = k;
        visitK(root);
        return num;
    }
    private void visitK(TreeNode root){
        if(root == null){
            return;
        }
        if(root.left != null){
            visitK(root.left);
        }
        remain--;
        if(remain == 0){
            num = root.val;
            return;
        }
        if(root.right != null){
            visitK(root.right);
        }
    }

你可能感兴趣的:(LeetCode)