面试题09. 用两个栈实现队列【LeetCode剑指offer】
程序员文章站
2022-06-17 17:47:04
...
题目:
思路
- 两个栈,一个size(队列中数据的数量)
- 入队列直接压入栈1,并且size++;
- 出队列时,先判断size是否为0,若size 为 0,则说明队列空,返回-1;
- 否则直接取出栈2的栈顶元素,若栈2为空,则把栈1 的所有元素一次性出栈压入栈2,此时栈2,按顺序弹出则是队列出队列的顺序,先进先出;出队列时,要相应的的size–;
实现:
class CQueue {
Stack<Integer> stack1;
Stack<Integer> stack2;
int size;
public CQueue() {
stack1 = new Stack();
stack2 = new Stack();
size = 0;
}
public void appendTail(int value) {
stack1.push(value);
size++;
}
public int deleteHead() {
if (size == 0) {
return -1;
}
if (stack2.empty()) {
while (!stack1.empty()) {
stack2.push(stack1.pop());
}
}
size--;
return stack2.pop();
}
}