155、最小栈

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/benzhaohao/article/details/86672851

设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。

  • push(x) -- 将元素 x 推入栈中。
  • pop() -- 删除栈顶的元素。
  • top() -- 获取栈顶元素。
  • getMin() -- 检索栈中的最小元素。

示例:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin();   --> 返回 -3.
minStack.pop();
minStack.top();      --> 返回 0.
minStack.getMin();   --> 返回 -2.

第一种思路:(只使用一个栈)

每次入栈2个元素,一个是入栈的元素本身,一个是当前栈元素的最小值。

class MinStack {
    //每次入栈两个元素,第一个元素是本身,另一个元素是当前栈的最小值
    public Stack<Integer> stack;
    /** initialize your data structure here. */
    public MinStack() {
     stack = new Stack<>();
}

public void push(int x) {
    if(stack.isEmpty()){
        stack.push(x);
        stack.push(x);
    }else{
        int tmp = stack.peek();
        stack.push(x);
        if(tmp<x){
            stack.push(tmp);
        }else{
            stack.push(x);
        }
    }
}

public void pop() {
    stack.pop();
    stack.pop();
}

public int top() {
    return stack.get(stack.size()-2);
}

public int getMin() {
    return stack.peek();
}
}

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack obj = new MinStack();
 * obj.push(x);
 * obj.pop();
 * int param_3 = obj.top();
 * int param_4 = obj.getMin();
 */

第二种思路:(使用两个栈)

一个栈存放入栈元素,另一个栈存放栈最小元素。

          class Minstack1{
		//设计一个辅助栈来保存栈中最小元素
		public Stack<Integer> stack;
		public Stack<Integer> minstack;
		public MinStack() {
			stack=new Stack<>();
			minstack=new Stack<>();
		}
		public void push(int x) {
			if(minstack.isEmpty()||x<minstack.peek())
				minstack.push(x);
			stack.push(x);
		}
		public void pop() {
			if(stack.peek().equals(minstack.peek()))
				minstack.pop();
			stack.pop();
		}
		public int top() {
			return stack.peek();
		}
		public int getMin() {
			return minstack.peek();
		}
	}

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack obj = new MinStack();
 * obj.push(x);
 * obj.pop();
 * int param_3 = obj.top();
 * int param_4 = obj.getMin();
 */

猜你喜欢

转载自blog.csdn.net/benzhaohao/article/details/86672851