栈、队列



Stack先进后出,常用函数3P: peek push pop

栈、队列_第1张图片
Stack

Queue先进先出,常用函数POP: peek offer poll
注意,Stack是一个Class,而Queue在Java里是interface,一般用LinkedList实现。


栈、队列_第2张图片
Queue


1. 最小栈

设计含返回栈中最小数的函数的栈,要求时间复杂度为O(1)。
做法是使用一个记录最小值的栈,每层的元素对应原栈相同层截止最小的元素。

public class MinStack {
    
    Stack stack; 
    Stack min; 
    

    /** initialize your data structure here. */
    public MinStack() {
        stack = new Stack();
        min = new Stack();
    }
    
    public void push(int x) {
        if(stack.empty() || x < min.peek()){
            min.push(x);
        }else{
            min.push(min.peek());
        }
        stack.push(x);
        
    }
    
    public void pop() {
        stack.pop();
        min.pop();
    }
    
    public int top() {
        return stack.peek();
    }
    
    public int getMin() {
        return min.peek();
    }
}


2. 两个栈实现一个队列

让一个栈实现队列的顺序,就是在每插入一个元素时,把元素插到栈底,即为队尾。这时候就需要用到一个辅助栈来记录原栈成员,实现原栈的修复。

如果一个栈的元素一个个跳出同时另一个栈不停把元素push进去,那么另一个栈的跳出顺序就会反过来,就会变成队列的顺序,而另一个栈再进行一个个跳出让原栈一个个push,原栈的顺序变回跟以前一样。

还有一个类似的思路是,push不变,原栈还是保持栈的顺序,只是跳出时跳栈底的元素。

public class MyQueue {
    
    Stack queue;

    /** Initialize your data structure here. */
    public MyQueue() {
        queue = new Stack();
    }
    
    /** Push element x to the back of queue. */
    public void push(int x) {
        Stack temp = new Stack();
        
        while(!queue.empty()){
            temp.push(queue.pop());
        }
        temp.push(x);
        while(!temp.empty()){
            queue.push(temp.pop());
        }
        
    }
    
    /** Removes the element from in front of queue and returns that element. */
    public int pop() {
        return queue.pop();
        
    }
    
    /** Get the front element. */
    public int peek() {
        return queue.peek();
    }
    
    /** Returns whether the queue is empty. */
    public boolean empty() {
        return queue.empty();
    }
}


3. 两个队列实现一个栈
public class MyStack {
    
    Queue stack;

    /** Initialize your data structure here. */
    public MyStack() {
        stack = new LinkedList();  
    }
    
    /** Push element x onto stack. */
    public void push(int x) {
        Queue temp = new LinkedList();
        
        while(stack.peek() != null){
            temp.offer(stack.poll());
        }
        stack.offer(x);
        while(temp.peek() != null) {
            stack.offer(temp.poll());
        }
        
    }
    
    /** Removes the element on top of the stack and returns that element. */
    public int pop() {
        return stack.poll();
    }
    
    /** Get the top element. */
    public int top() {
        return stack.peek();
    }
    
    /** Returns whether the stack is empty. */
    public boolean empty() {
        return stack.peek() == null;
    }
}


4. 出栈入栈合理性

输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。
这里就用一个栈来储存和模拟压入和弹出的过程。按照压入顺序依次压入元素进栈,如果发现栈顶元素刚好是当前弹出元素,说明此时发生一次弹出,栈弹出一个元素且弹出数组后移一位。整个过程结束后(遍历完pushA后),如果栈为空且popA也遍历完,则说明合理;反之不合理。
(P.S.之前代码思路有误但牛客网上仍然通过,说明牛客网测试用例设计得不够全面。幸得朋友指出。)

 public boolean IsPopOrder(int [] pushA,int [] popA) {
     
        if(pushA == null || popA == null) return false;
        Stack stack = new Stack();
     
        int j = 0;
        for(int i = 0; i < pushA.length; i++){
            stack.push(pushA[i]);
            while(!stack.isEmpty() && stack.peek() == popA[j]){
                stack.pop();
                j++;
            }           
        }     
        return (stack.isEmpty() && popA.length == j);
    }

你可能感兴趣的:(栈、队列)