力扣剑指 Offer 09. 用两个栈实现队列
题目:
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
解题:
- 初始化两个栈:把两个栈的数据全部出栈清空;
- 添加数据:栈1的入口如同队列的尾部
在栈1中直接入栈 - 删除数据:栈2的出口如同队列的头部
如果栈2有数据,就直接输出栈2顶部的元素
如果栈2没有数据,看栈1有没有数据,如果有就导入栈2再输出栈2顶部的元素,如果没有就输出-1
class CQueue {
// 定义两个栈
stack<int> stack1,stack2;
public:
// 初始化两个栈为空
CQueue() {
while (!stack1.empty()) {
stack1.pop();
}
while (!stack2.empty()) {
stack2.pop();
}
}
// 添加数直接入栈栈1
void appendTail(int value) {
stack1.push(value);
}
// 删除队头需要分情况
int deleteHead() {
// 如果栈2为空,把栈1的数据全部放到栈2
if (stack2.empty()) {
while (!stack1.empty()) {
stack2.push(stack1.top());
stack1.pop();
}
}
// 如果栈2还为空,返回-1
if (stack2.empty()) {
return -1;
} else { // 如果栈2有数据,则输出栈顶
int deleteItem = stack2.top();
stack2.pop();
return deleteItem;
}
}
};
剑指 Offer 30. 包含min函数的栈
题目:
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
解题:
分两个栈做此题:
一个为数据栈,所有数据入栈出栈都是正常出入;
另一个为最小值栈,栈顶放的是当前数据栈的最小值,每次入栈都要进行比较,比最小值栈的栈顶小才能入栈。
class MinStack {
stack<int> dataStack;
stack<int> minStack;
public:
/** initialize your data structure here. */
MinStack() {
while (!dataStack.empty()) {
dataStack.pop();
}
while (!minStack.empty()) {
minStack.pop();
}
}
void push(int x) {
dataStack.push(x);
if(minStack.empty() || x<=minStack.top()){
minStack.push(x);
}
}
void pop() {
if(dataStack.top()==minStack.top()){
dataStack.pop();
minStack.pop();
}else{
dataStack.pop();
}
}
int top() {
return dataStack.top();
}
int min() {
return minStack.top();
}
};