题目描述
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof/
思路
栈是先进后出,队列是先进先出,即用两个栈实现先进先出,包括压入和弹出操作。
- push ,直接将value压进栈里,会导致该元素后进先出,所以我们应该将其压入栈底。考虑s1已经是倒序压入后形成的栈,只需将s1中的元素push进s2,直至s1为空,这样再将value 压进s1,就实现了将value压进s1底部的操作,但此时其他元素还在s2内,需要将他们再一个个push进s1,直至s2为空;
- top 因为在push操作中我们已经实现将新的元素push进栈底,所以栈顶元素一定是年纪最大的元素,直接用栈的top操作即可,别忘了pop哟~;
c++代码
class CQueue {
public:
stack<int> s1,s2;
CQueue() {
}
void appendTail(int value) {
if(s1.empty()) s1.push(value);
else {
while(!s1.empty()){
s2.push(s1.top());
s1.pop();
}
s1.push(value);
while(!s2.empty()){
s1.push(s2.top());
s2.pop();
}
}
}
int deleteHead() {
if(s1.empty()) return -1;
int res=s1.top();
s1.pop();
return res;
}
};