题目
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push
、pop
、peek
、empty
):
实现 MyQueue
类:
void push(int x)
将元素 x 推到队列的末尾int pop()
从队列的开头移除并返回元素int peek()
返回队列开头的元素boolean empty()
如果队列为空,返回true
;否则,返回false
说明:
-
你 只能 使用标准的栈操作 ------ 也就是只有
push to top
,peek/pop from top
,size
, 和is empty
操作是合法的。 -
你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的
-
栈操作即可。
-
1 <= x <= 9
-
最多调用
100
次push
、pop
、peek
和empty
-
假设所有操作都是有效的 (例如,一个空的队列不会调用
pop
或者peek
操作)
题解
栈的实现
代码
//开辟、初始化
MyQueue* myQueueCreate() {
MyQueue* obj = (MyQueue*)malloc(sizeof(MyQueue));
STInit(&obj->push_st); //不要忘记取地址&
STInit(&obj->pop_st);
return obj;
}
void myQueuePush(MyQueue* obj, int x) {
STPush(&obj->push_st, x);
}
int myQueuePop(MyQueue* obj) {
int pop = myQueuePeek(obj);
//由于调用了myQueuePeek函数,此时pop栈一定不为空
STPop(&obj->pop_st);
return pop;
}
int myQueuePeek(MyQueue* obj) {
if (STEmpty(&obj->pop_st)) //如果popst(是obj的结构成员)为空
{
while (!STEmpty(&obj->push_st)) //当不为空,持续进行
{
STPush(&obj->pop_st, STTop(&obj->push_st));
STPop(&obj->push_st);
}
}
return STTop(&obj->pop_st);
}
bool myQueueEmpty(MyQueue* obj) {
return STEmpty(&obj->push_st) && STEmpty(&obj->pop_st);
}
void myQueueFree(MyQueue* obj) {
STDestroy(&obj->pop_st);
STDestroy(&obj->push_st);
free(obj);
}