标签:包含min函数的栈
题目:定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的函数。在该栈中,调用min,push,pop的时间复杂度都为O(1)。
#include<iostream> #include<vector> using namespace std; template <class T> class Stack { public: void push(T data) { if(_array.size()==0) { _array.push_back(data); _array1.push_back(0); } else { int index=_array1.size()-1; if(data>=_array[index]) { _array.push_back(data); _array1.push_back(_array1[index]); } else { _array.push_back(data); _array1.push_back(index+1); } } } void pop() { int index=_array.size()-1; _array[index]=NULL; _array1[index]=NULL; } T Min() { int end=_array1.size()-1; int index=_array1[end]; return _array[index]; } private: vector<T> _array; vector<int> _array1; }; void test() { Stack<int> s1; s1.push(3); s1.push(7); s1.push(4); s1.push(1); s1.push(8); s1.push(0); cout<<s1.Min()<<endl; } int main() { test(); system("pause"); return 0; }
结果:
本文出自 “liveyoung” 博客,转载请与作者联系!
标签:包含min函数的栈
原文地址:http://youngyoungla.blog.51cto.com/10697042/1773644