地址:
力扣https://leetcode-cn.com/problems/implement-queue-using-stacks/
题目:
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(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(双端队列)来模拟一个栈,只要是标准的栈操作即可。
进阶:
你能否实现每个操作均摊时间复杂度为 O(1) 的队列?换句话说,执行 n 个操作的总时间复杂度为 O(n) ,即使其中一个操作可能花费较长时间。
示例:
输入: 解释: |
提示:
1 <= x <= 9 最多调用 100 次 push、pop、peek 和 empty 假设所有操作都是有效的 (例如,一个空的队列不会调用 pop 或者 peek 操作) |
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/implement-queue-using-stacks
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
队列是一种 先进先出(FIFO) 的模式
栈 是一种 先进后出(FOFI) 的模式
这两种模式正好相反,要用栈来实现队列,就要用到 两次出入栈,这样才能负负得正
示例:
我们用两个栈来实现队列,分别为 入口栈 和 出口栈
数据 1,2,3,4 准备阶段
入队操作 就是对数据进行压栈到入口栈的过程
出队操作 是对 出口栈 进行出栈操做,如果出口栈没有数据,那么先要将 入口栈数据压栈到出口栈
出口栈有数据后的出栈
这样就完成数据的 入队+出队
另外,由于 C 没有提供现成的 队列,栈 API,所以这部分需要自行实现
对于初阶人员,可能会比较模糊,我们可以将
栈 操作依序完成,再来完成题目提供的待填 队列 API,就比较简单了
方法一、双栈操作
typedef struct {
int *stk;
int stkSize;
int stkCapacity;
} MyStack;
typedef struct {
MyStack *inStk;
MyStack *outStk;
} MyQueue;
MyStack *myStackCreate(int capacity)
{
MyStack *stack = (MyStack *)malloc(sizeof(MyStack));
stack->stk = (int *)malloc(sizeof(int) * capacity);
stack->stkSize = 0;
stack->stkCapacity = capacity;
return stack;
}
void myStackFree(MyStack *stack)
{
if(stack)
{
free(stack->stk);
free(stack);
}
}
bool myStackEmpty(MyStack *stack) {
return stack->stkSize == 0;
}
void myStackPush(MyStack *stack, int x)
{
stack->stk[stack->stkSize++] = x;
}
void myStackPop(MyStack *stack)
{
stack->stkSize--;
}
int myStackTop(MyStack *stack)
{
return stack->stk[stack->stkSize - 1];
}
void in2out(MyQueue *queue)
{
while(! myStackEmpty(queue->inStk))
{
int x = myStackTop(queue->inStk);
myStackPop(queue->inStk);
myStackPush(queue->outStk, x);
}
}
MyQueue* myQueueCreate() {
MyQueue *queue = (MyQueue *)malloc(sizeof(MyQueue));
queue->inStk = myStackCreate(100);
queue->outStk = myStackCreate(100);
return queue;
}
void myQueuePush(MyQueue* obj, int x) {
myStackPush(obj->inStk, x);
}
int myQueuePop(MyQueue* obj) {
if(myStackEmpty(obj->outStk)) // outStk is empty, push inStk to outStk first
{
in2out(obj);
}
int x = myStackTop(obj->outStk);
myStackPop(obj->outStk);
return x;
}
int myQueuePeek(MyQueue* obj) {
if(myStackEmpty(obj->outStk)) // outStk is empty, push inStk to outStk first
{
in2out(obj);
}
return myStackTop(obj->outStk);
}
bool myQueueEmpty(MyQueue* obj) {
return myStackEmpty(obj->inStk) && myStackEmpty(obj->outStk);
}
void myQueueFree(MyQueue* obj) {
if(obj)
{
myStackFree(obj->inStk);
myStackFree(obj->outStk);
}
}
/**
* Your MyQueue struct will be instantiated and called as such:
* MyQueue* obj = myQueueCreate();
* myQueuePush(obj, x);
* int param_2 = myQueuePop(obj);
* int param_3 = myQueuePeek(obj);
* bool param_4 = myQueueEmpty(obj);
* myQueueFree(obj);
*/