题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
解题思路
栈:先进后出
队列:先进先出
入队:入栈1
如 1,2依次入队
出队:
若栈2为空,则栈1元素依次出栈,放入栈2,栈2出栈
若栈2不为空,则栈2元素出栈。
执行出队操作,由于1比2先进队,出队时实现1先出队,所以需要将栈1元素依次放入栈2 中,即栈2为空时,出队操作为栈2出栈。
此时栈2中只有2,执行操作 3入队
再次执行出队操作,由于2比3先入队,我们想实现的是2出队,所以栈2不空时,出队操作即栈2出栈
import java.util.Stack;
public class Solution {
Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
public void push(int node) {
//入队操作为栈1入栈
stack1.push(node);
}
public int pop(){
//栈2不空,出队操作:栈2出栈
if(!stack2.empty())
{
return stack2.pop();
}
//栈2为空时,进队操作:栈1元素依次出栈入栈2,全部入栈2后,栈2出栈
while(!stack1.empty()){
stack2.push(stack1.pop());
}
return stack2.pop();
}
}