牛客-NC76-用两个栈实现队列
NC76. 用两个栈实现队列(easy)
方法一:直接模拟法(自己写的)
思路:stack1拿来存值,stack2拿来取值。push方法非常直接,稍微有点曲折的是pop方法,我们需要做判断:(1)stack2还有上一次stack1倒过来的值,而此时pop只需要将stack2中的栈顶元素pop出来就行。(2)当stack2为空,此时需要将stack1新增的值倒过来,再pop就行了。
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 拿来存值
stack1.push(node);
}
public int pop() {
// stack2拿来取值
// 判断stack2中是否有值
if (!stack2.isEmpty()) {
return stack2.pop();
} else {
// 从stack1中导值
while (!stack1.isEmpty()) {
stack2.push(stack1.pop());
}
}
return stack2.pop();
}
}
时间复杂度: O(N), N是存放到stack1中的元素,因为极端情况下,可能将N个元素一次性push到stack1,再一次性从stack2中pop出来。
空间复杂度: O(1), 未使用额外的空间。
总结:上一个题解才说到栈实现队列,这就遇上了。盘它,详情请看link。