LintCode-11.二叉查找树中搜索区间

题目

描述

实现一个带有取最小值min方法的栈，min方法将返回当前栈中的最小值。

你实现的栈将支持push，pop 和 min 操作，所有操作要求都在O(1)时间内完成。

样例

如下操作：push(1)，pop()，push(2)，push(3)，min()， push(1)，min() 返回 1，2，1

解答

思路

建立两个栈，一个保持顶端是最小的数，另一个保存剩下的数据。

代码

/**
 * Definition of TreeNode:
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left, right;
 *     public TreeNode(int val) {
 *         this.val = val;
 *         this.left = this.right = null;
 *     }
 * }
 */
public class Solution {
    /**
     * @param root: The root of the binary search tree.
     * @param k1 and k2: range k1 to k2.
     * @return: Return all keys that k1<=key<=k2 in ascending order.
     */
    ArrayList result = new ArrayList();
    public ArrayList searchRange(TreeNode root, int k1, int k2) {
        // write your code here
        if(root == null) return result;
        if(root.val >= k1 && root.val <= k2){
            result.add(root.val);
            searchRange(root.left, k1, k2);
            searchRange(root.right, k1, k2);
        }
        else if(root.val < k1){
            searchRange(root.right, k1, k2);
        }
        else if(root.val > k2){
            searchRange(root.left, k1, k2);
        }
        Collections.sort(result);
        return result;
    }
}