LeetCode题解-230-Kth Smallest Element in a BST

原题

概览

BST是二叉查找树，它具有如下的性质：左子树中所有节点的值均小于根节点的值，右子树中所有节点的值均大于根节点的值。

解法1使用了迭代法；

解法2使用了递归法。

解法1

解题思路

使用迭代法。由于BST的性质，使用二叉树的中序遍历。中序遍历先访问到的节点的值较小，每访问一个节点计数值+1，当计数值到达K的时候该节点就是要访问的节点。

中序遍历的思路请见： LeetCode题解-94-Binary Tree Inorder Traversal

代码

public class Solution230_iterator {
    public int kthSmallest(TreeNode root, int k) {
        int count = 1;
        Stack<TreeNode> stack = new Stack<TreeNode>();
        stack.push(root);

        while (!stack.empty()){
            while (stack.peek() != null){
                TreeNode currentNode = stack.peek();
                stack.push(currentNode.left);
            }
            stack.pop();

            if (!stack.isEmpty()){
                TreeNode currentNode = stack.pop();
                if (k == count++)
                    return currentNode.val;

                stack.push(currentNode.right);
            }
        }
        return -1;
    }
}

解法2

解题思路

递归法，仍然是中序遍历。参考了 3 ways implemented in JAVA: Binary Search, in-order iterative & recursive public int kthSmallest(TreeNode root, int k) 

自己没有想出来递归方法，因为觉得 public int kthSmallest(TreeNode root, int k) 这样的函数不好构造递归。但是没有想到可以在构造一个新的函数，在 public int kthSmallest(TreeNode root, int k) 中调用该函数。

代码

public class Solution230_recursive {
    private static int count = 0;
    private static int number = 0;

    public int kthSmallest(TreeNode root, int k) {
        count = k;
        helper(root);
        return number;
    }

    private void helper(TreeNode root) {
        if (root.left != null)
            helper(root.left);

        count--;
        if (count == 0) {
            number =  root.val;
            return;
        }
        if (root.right != null)
            helper(root.right);
    }
}