算法小白开始刷题,本文包含了自己的思考过程,所以内容可能比较冗余,如需思路,可直接看文章后面的 优化解法。
题目链接:https://leetcode-cn.com/problems/implement-queue-using-stacks-lcci/
题目
实现一个MyQueue类,该类用两个栈来实现一个队列。
示例:
MyQueue queue = new MyQueue();
queue.push(1);
queue.push(2);
queue.peek(); // 返回 1
queue.pop(); // 返回 1
queue.empty(); // 返回 false
说明:
- 你只能使用标准的栈操作 – 也就是只有 push to top, peek/pop from top, size 和 is empty 操作是合法的。
- 你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。
- 假设所有操作都是有效的 (例如,一个空的队列不会调用 pop 或者 peek 操作)。
思路
同之前的用两个栈实现队列相同。设置两个栈 stack1、stack2,入队操作即是将元素 push 进栈 stack1 中,执行 peek 操作和 pop 操作时,先 pop stack2 中的元素,若是 stack2 为空,则将 stack1 中的元素全部 push 进 stack2 中,再读取 stack2 中的 top 元素。
代码
class MyQueue {
std::stack<int> stack1, stack2;
public:
/** Initialize your data structure here. */
MyQueue() {
while(!stack1.empty()) {
stack1.pop();
}
while(!stack2.empty()) {
stack2.pop();
}
}###
/** Push element x to the back of queue. */
void push(int x) {
stack1.push(x);
}
/** Removes the element from in front of queue and returns that element. */
int pop() {
int result;
if(!stack2.empty()) {
result = stack2.top();
stack2.pop();
}
else {
if(stack1.empty())
return -10000;
else {
while(!stack1.empty()) {
stack2.push(stack1.top());
stack1.pop();
}
result = stack2.top();
stack2.pop();
}
}
return result;
}
/** Get the front element. */
int peek() {
int result;
if(!stack2.empty())
result = stack2.top();
else {
if(stack1.empty())
return -10000;
else {
while(!stack1.empty()) {
stack2.push(stack1.top());
stack1.pop();
}
result = stack2.top();
}
}
return result;
}
/** Returns whether the queue is empty. */
bool empty() {
if(stack1.empty() && stack2.empty())
return true;
else
return false;
}
};
/**
* Your MyQueue object will be instantiated and called as such:
* MyQueue* obj = new MyQueue();
* obj->push(x);
* int param_2 = obj->pop();
* int param_3 = obj->peek();
* bool param_4 = obj->empty();
*/
成绩
优化解法
pop 函数和 peek 函数 还是写的有些复杂,可以再进行简化:
/** Removes the element from in front of queue and returns that element. */
int pop() {
int result;
if(stack2.empty()) {
while(!stack1.empty()) {
stack2.push(stack1.top());
stack1.pop();
}
}
result = stack2.top();
stack2.pop();
return result;
}
/** Get the front element. */
int peek() {
int result;
if(stack2.empty()) {
while(!stack1.empty()) {
stack2.push(stack1.top());
stack1.pop();
}
}
result = stack2.top();
return result;
}