#include "stdafx.h"
#include <iostream>
#include<stdio.h>
#include <fstream>
#include <sstream>
#include <cstring>
#include <iterator>
using namespace std;
#include <vector>
#include <algorithm>
#include <Windows.h>
// BSTreeNode1.cpp : 定义控制台应用程序的入口点。
//
template <class T>
class StackWithMin
{
public:
void push(const T &data)
{
data_stack.push_back(data);
if (0==min_stack.size())
{
min_stack.push_back(data);
}
else if (min_stack[min_stack.size()-1]>data)
{
min_stack.push_back(data);
}
}
void pop()
{
assert(data_stack.size());
if (data_stack.size())
{
if (data_stack[data_stack.size()-1]==min_stack[min_stack.size()-1])
{
min_stack.pop_back();
}
data_stack.pop_back();
}
}
T& min1()
{
return min_stack[min_stack.size() - 1];
}
T& top() //返回最上面的值
{
return data_stack[data_stack.size()-1];
}
private:
vector<T> data_stack; //使用向量 Vector 来存储数据
vector<T> min_stack;
};
int main()
{
StackWithMin<int> stack;
int a[5]={10,2,4,23,3};
for (int i=0;i<5;i++)
{
stack.push(a[i]);
}
int a1=stack.min1();
cout<<"最小值"<<a1;
system("pause");
return 0;
}
定义栈的数据结构,要求添加一个 min 函数,能够得到栈的最小元素。要求函数 min、push 以及 pop 的时间复杂度都是 O(1)。
最新推荐文章于 2020-07-22 10:46:34 发布