来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/implement-queue-using-stacks
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
232.请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(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(双端队列)来模拟一个栈,只要是标准的栈操作即可。
- 思想其实不难,利用两个栈来实现队列,一个栈用来进元素,另一个栈用来出元素。但是编写这么多的代码难免会出带点差错,还是动手编写一下吧。由于题目并没有限定队列的大小是多少,因此没有对队列满的情况进行判断,如果有限定大小的话,代码还需要进一步改进。
typedef struct Stack {
int *stack; //栈的存储空间
int top; //指向栈顶元素的指针
int capacity; //栈的容量
} Stack;
Stack* stackCreate(int capacity) {
Stack *mstack = (Stack*)malloc(sizeof(Stack));
mstack->top = -1;
mstack->capacity = capacity;
mstack->stack = (int*)malloc(sizeof(int) * capacity);
return mstack;
}
void stackPush(Stack* obj, int x) {
obj->stack[++(obj->top)] = x;
}
int stackPop(Stack* obj) {
return obj->stack[(obj->top)--];
}
int stackPeek(Stack* obj) {
return obj->stack[obj->top];
}
bool isStackEmpty(Stack* obj) {
return obj->top == -1;
}
void stackFree(Stack* obj) {
free(obj->stack);
}
typedef struct {
Stack *inStack;
Stack *outStack;
} MyQueue;
/** Initialize your data structure here. */
MyQueue* myQueueCreate() {
MyQueue *mQueue = (MyQueue*)malloc(sizeof(MyQueue));
mQueue->inStack = stackCreate(100);
mQueue->outStack = stackCreate(100);
return mQueue;
}
void in2out(MyQueue* obj) {
while ( !isStackEmpty(obj->inStack)) {
stackPush(obj->outStack, stackPop(obj->inStack));
}
}
/** Push element x to the back of queue. */
void myQueuePush(MyQueue* obj, int x) {
stackPush(obj->inStack, x);
}
/** Removes the element from in front of queue and returns that element. */
int myQueuePop(MyQueue* obj) {
if (isStackEmpty(obj->outStack))
in2out(obj);
return stackPop(obj->outStack);
}
/** Get the front element. */
int myQueuePeek(MyQueue* obj) {
if (isStackEmpty(obj->outStack))
in2out(obj);
return stackPeek(obj->outStack);
}
/** Returns whether the queue is empty. */
bool myQueueEmpty(MyQueue* obj) {
return isStackEmpty(obj->inStack) && isStackEmpty(obj->outStack);
}
void myQueueFree(MyQueue* obj) {
stackFree(obj->inStack);
stackFree(obj->outStack);
}
/**
* 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);
*/