题目
描述
实现一个带有取最小值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<Integer> result = new ArrayList<Integer>();
public ArrayList<Integer> 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;
}
}