java两个堆栈实现队列

package exercise1;

import java.util.Stack;

public class Solution {

    Stack<Integer> stack1 = new Stack<Integer>();

    Stack<Integer> stack2 = new Stack<Integer>();

    

    public void push(int node) {

        stack1.push(node); 

    }

    

    public int pop() {

while(!stack1.empty()){

stack2.push(stack1.pop());

}

int first = stack2.pop();

while(!stack2.empty()){

stack1.push(stack2.pop());

}

return first;

    }

    public static void main(String[] args) {

Solution solution=new Solution();

solution.push(1);

solution.push(2);

solution.push(3);

solution.push(4);

System.out.println(solution.pop());

solution.push(2);

System.out.println(solution.pop());

System.out.println(solution.pop());

System.out.println(solution.pop());

System.out.println(solution.pop());

}

}

很容易想到把数据压入到第一个栈然后弹出到第二个栈,这时弹出第二个栈就能实现先进先出了。

这里pop里肯定要这么写:

while(!stack1.empty()){

stack2.push(stack1.pop());

}

但是之后不能就不管了直接return stack.pop();

因为如果弹出一个后再压入一个元素,是压入stack1中,而此时1已经空了,会把元素压入到2的顶端,从而出问题。

所以stack1肯定是要维护着的,而stack2只要1的尾端的值弹出就好了,剩下的值重新给回stack1,以便新的值压入。

点赞