使用C++完成《程序员代码面试指南 IT名企算法与数据结构题目》
题目
实现一个特殊的栈,在实现栈的基本功能基础上,再实现返回栈中最小元素的操作
实现
使用两个栈,一个栈用来保存当前栈中的元素,起功能和一个正常的栈没有区别,这个栈记为stackData;另一个栈用于保存每一步的最小值,这个栈记为stackMin
1. 第一种方案
压入数据规则:
假设当前数据为newNum,先将其压入stackData,然后判断stackMin是否为空:
- 如果为空,则newNum压入stackMin
- 如果不为空,则比较newNum和stackMin的栈顶元素中哪一个更小
- 如果newNum更小或者两者相等,则newNum也压入stackMin
- 如果stackMin中栈顶元素小,则stackMin不压入任何内容
弹出数据规则
- 先在stackData中弹出栈顶元素,记为value。然后比较当前stackMin的栈顶元素和value哪一个更小
- 当value等于stackMin的栈顶元素时,stackMin弹出栈顶元素
- 当value大于stackMin的栈顶元素时,stackMin不弹出栈顶元素:返回value
getMin操作(查询当前栈中的最小值操作)
由于stackMin始终记录着stackData中的最小值,所以,stackMin的栈顶元素始终是当前stackData中的最小值
方案1实现代码
class MyStack1{
private:
stack <int>stackData;
stack <int>stackMin;
public:
MyStack1(){}
void push(int newNum){
if(stackMin.empty()){
stackMin.push(newNum);
}else if (newNum <= getmin()){
stackMin.push(newNum);
}
stackData.push(newNum);
}
int pop(){
if(stackData.empty()){
fprintf(stderr, "The stcack is empty\n");
exit(0);
}
int value = stackData.top();
stackData.pop();
if(value == getmin()){
stackMin.pop();
}
return value;
}
int getmin(){
if(stackMin.empty()){
fprintf(stderr,"Your Stack is empty");
exit(0);
}
return stackMin.top();
}
};
2. 第二种方案
压入数据规则
- 假设当前数据为newNum,先将其压入stackData,然后判断stackMin是否为空:
- 如果为空,则newNum也压入stackMin
- 如果不为空,则比较newNum和stackMin的栈顶元素中哪一个更小:
- 如果newNum更小或两者相等,则newNum也压入stackMin
- 如果stackMin中栈顶元素小,则把stackMin的栈顶元素重复压入stackMin
弹出数据规则
- 在stackData中弹出数据,弹出的数据记为value;
- 弹出stackMin中的栈顶;
- 返回value
getMin操作(查询当前栈中最小值操作)
由于stackMin始终记录着stackData中的最小值,所以,stackMin的栈顶元素即为当前stackData中的最小值
方案2实现代码
class MyStack2{
private:
stack <int>stackData;
stack <int>stackMin;
public:
MyStack2(){}
void push(int newNum){
if(stackMin.empty()){
stackMin.push(newNum);
}else if (newNum<getmin()){
stackMin.push(newNum);
}else{
int newMin = stackMin.top();
stackMin.push(newMin);
}
stackData.push(newNum);
}
int pop(){
if(stackData.empty()){
fprintf(stderr,"The stack is empty");
exit(0);
}
stackMin.pop();
int popNum=stackData.top();
stackData.pop();
return popNum;
}
int getmin(){
if(stackMin.empty()){
fprintf(stderr, "Your stack is empty");
exit(0);
}
return stackMin.top();
}
};
第一种方案在stackMin的栈顶只存放当前stackData里的最小值一次,而第二种方案每次在压入和弹出stackData栈时都要压入和弹出stackMin中数据,向stackMin栈中重复压入当前stackData里的最小值