思路:
push:
利用两个栈,第一个栈顺序存放元素,第二个栈当为空时,先存入一个元素,再次存入时,如果要存放的元素比第二个栈顶的元素大,则不存入,小于或等于的时候存入,就能保证第二栈顶都是第一个栈中最小的元素
pop:
应该判断第一个栈顶的元素与第二个栈顶的元素是否相等,不等,则只删除栈一的元素,相等的时候,栈1与栈2 都删除,防止错删元素
#pragma once
#include<iostream>
#include<stack>
using namespace std;
typedef int T;
class CMyStack
{
public:
CMyStack()
{}
~CMyStack()
{}
void Push(T x)
{
s1.push(x);//先把数据放在栈1中
if (s2.empty() || x <= s2.top())
s2.push(x);
//如果栈2 为空或者栈2顶的元素大于x,把x放入栈2,始终保持栈2顶的元素最小
}
void Pop()
{
if (s1.top() == s2.top())
s2.pop();
//如果栈1 顶的元素等于栈2顶的元素,删除栈2顶元素再删除栈一顶元素,保证不会错删小元素
}
T Min()//如果栈2不为空,弹出栈2顶元素,则为最小元素
{
if (!s2.empty())
return s2.top();
else
cout<<"Mystack is empty"<<endl;
}
void Print()
{
while (!s1.empty()) //打印栈一元素
{
cout << s1.top() << "->";
s1.pop();
}
cout << endl;
while (!s2.empty()) //打印栈2元素
{
cout << s2.top() << "->";
s2.pop();
}
cout << endl;
}
private:
stack<T> s1, s2;
};