剑指Offer09.用两个栈实现队列
原题
根据栈先进后出和队列先进先出的不同特点予以以下实现,使用栈s1记录队列尾部元素,每次插入操作可以直接压入栈s1中;在删除元素时,为了删除头部元素,使用栈s2维护头部元素。当s2不为空时,直接弹出栈顶元素即可。当s2为空时,就将s1中元素一次弹出而后压入s2中,实现顺序的反转,s2栈顶元素就是队列的头部元素。当栈为空时,返回-1。
class CQueue {
stack<int> s1;//记录队列元素(栈顶为尾部)
stack<int> s2;//删除元素时使用(栈顶为头部)
public:
CQueue() {
}
void appendTail(int value) {
s1.push(value);
}
int deleteHead() {
if(s2.empty()) {
while(!s1.empty()) {
auto it=s1.top();
s1.pop();
s2.push(it);
}
}
if(s2.empty()) return -1;
else {
auto it=s2.top();
s2.pop();
return it;
}
}
};
/**
* Your CQueue object will be instantiated and called as such:
* CQueue* obj = new CQueue();
* obj->appendTail(value);
* int param_2 = obj->deleteHead();
*/
插入和删除元素的时间复杂度均为O(1),看起来删除操作的时间复杂度是O(n),但由于每个元素至多被压入和弹出s2一次,故其时间复杂度为O(1)。空间复杂度为O(n)。