一、用两个栈实现队列
剑指 Offer 09. 用两个栈实现列https://leetcode.cn/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof/
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
示例 1:
输入:
["CQueue","appendTail","deleteHead","deleteHead","deleteHead"]
[[],[3],[],[],[]]
输出:[null,null,3,-1,-1]
示例 2:
输入:
["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]
[[],[],[5],[2],[],[]]
输出:[null,-1,null,null,5,2]
提示:
1 <= values <= 10000
最多会对 appendTail、deleteHead 进行 10000 次调用
解题思路:
1.栈特点: 先进后出。
2.队列特点:先进先出 。
3.两个栈实现队列:栈A负责入队,栈B负责将栈A数据倒叙然后出队。
具体细节:
1.入队直接将数据加入栈A。
2.出队时栈B不空则直接返回栈B栈顶元素。
3.出队时栈B为空且栈A也为空时返回-1。
4.出队时栈B为空栈A不为空则将栈A中所有元素出栈加入到栈B中实现倒叙,再返回栈B栈顶元素。
复杂度分析:
1.时间复杂度:入队为O(1),出队为O(N)。
2.空间复杂度:O(N),栈A,栈B共同保存N个元素。
代码实现:
class CQueue {
stack<int> stackA, stackB;
public:
CQueue() {
while (!stackA.empty()) {
stackA.pop();
}
while (!stackB.empty()) {
stackB.pop();
}
}
void appendTail(int value) {
stackA.push(value);
}
int deleteHead() {
if (!stackB.empty()) {
int deleteItem = stackB.top();
stackB.pop();
return deleteItem;
} else {
if (stackA.empty()) return -1;
else {
while (!stackA.empty()) {
stackB.push(stackA.top());
stackA.pop();
}
int deleteItem = stackB.top();
stackB.pop();
return deleteItem;
}
}
}
};
/**
* Your CQueue object will be instantiated and called as such:
* CQueue* obj = new CQueue();
* obj->appendTail(value);
* int param_2 = obj->deleteHead();
*/
二、包含min函数的栈
剑指 Offer 30. 包含min函数的栈https://leetcode.cn/problems/bao-han-minhan-shu-de-zhan-lcof/
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.
提示:
各函数的调用总次数不超过 20000 次
解题思路:
1.栈特点:先进后出,找到最小值需遍历一遍栈,时间复杂度O(N)。
2.用空间换时间可将时间复杂度缩减为O(1):栈A存数据,栈B存每次入栈的最小元素。
具体细节:
1.入栈时直接压入栈A,同时判断栈B栈顶元素,若栈B为空直接压入,非空则判断是否小于栈顶,若小于则压入栈B。
2.出栈时直接弹出栈A栈顶元素,若与栈B栈顶元素相同,则将栈B栈顶元素也删除。
3.返回栈顶元素直接返回栈A的栈顶元素。
4.返回最小元素则直接返回栈B的栈顶元素。
复杂度分析:
1.时间复杂度:都为O(1)。
2.空间复杂度:栈A需N,栈B最坏需要N。综合需要O(N)。
代码实现:
class MinStack {
public:
/** initialize your data structure here. */
stack<int> A;
stack<int> B;
MinStack() {
while (!A.empty()) {
A.pop();
}
while (!B.empty()) {
B.pop();
}
}
void push(int x) {
A.push(x);
if(B.empty()){
B.push(x);
}else if(B.top()>=x){
B.push(x);
}
}
void pop() {
if(A.top()==B.top()){
B.pop();
}
A.pop();
}
int top() {
return A.top();
}
int min() {
return B.top();
}
};