==> 学习汇总(持续更新)
==> 从零搭建后端基础设施系列(一)-- 背景介绍
题目:
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型
这个题目简单,但是做得太快可能会掉坑里。
方法一:
如果是第一次做,可能正常的思路就是用栈2当做交换器,push操作存入栈1,pop的时候将栈1导入栈2,弹出后,再导回去。
//push O(1)
public void push(int node) {
stack1.push(node);
}
//pop 恒定 O(2n)
public int pop() {
while (!stack1.empty()) {
stack2.push(stack1.pop());
}
int e = stack2.pop();
while (!stack2.empty()) {
stack1.push(stack2.pop());
}
return e;
}
方法二:
将两个栈连接成一个队列,栈1的头是队尾,底是队首,栈2的底是队尾,头是队首。
例如
进栈1 2 3,
栈1有
3
2
1
出栈1,栈2有
2
3
进栈4,栈1有
4
进栈5,栈1有
5
4
出栈2 3,栈2为空
出栈4,栈2有
5
//push O(1)
public void push2(int node) {
stack1.push(node);
}
//pop O(1) ~ O(n)
public int pop2() {
if(!stack2.empty()) {
return stack2.pop();
}
while (!stack1.empty()) {
stack2.push(stack1.pop());
}
return stack2.pop();
}
速度测试
使用random.nextBoolean()模拟push和pop操作。
方法/操作次数 | 100000 | 1000000 |
---|---|---|
方法1 | 1685 ms | 36532 ms |
方法2 | 16 ms | 82 ms |