您的位置:首页 > Web前端

剑指offer-21:包含min函数的栈

2017-09-08 10:46 197 查看
定义栈的数据结构,在该类型中实现一个能够得到栈的最小元素的min函数,该栈中调用min, push, pop的时间复杂度都是O(1)。

分析

要保持栈原本性质,就需要一个栈A存放原始数据

要得到栈当前最小元素,需要一个辅助栈B,专门存储当前的最小值

A中压栈,B中压当前的最小值(要么是新的最小,要么是B栈顶最小)

代码:

// offer-21-stack_min.cpp : 定义控制台应用程序的入口点。
//

//#include "stdafx.h"
#include <iostream>
#include <stack>

using namespace std;

// 定义一个类,表示新栈类型
class StackMin {
public:
//压栈
void push(int value) {
// 数据栈压栈
m_data.push(value);
// 辅助栈压入新的最小
if(m_min.size()==0 || m_min.top() > value )
m_min.push(value);
// 辅助栈压入原最小,仍为当前最小
else
m_min.push(m_min.top());
}
// 弹栈
void pop() {
m_data.pop();
m_min.pop();
}
// 获取当前栈顶,只和数据栈相关
int top() {
return m_data.top();
}
// 获取最小,只和辅助栈相关
int min() {
return m_min.top();
}
private:
stack<int> m_data;  // 数据栈
stack<int> m_min; // 存当前最小元素的栈
};

// 测试
int _tmain(int argc, _TCHAR* argv[])
{
Solution a;
a.push(3);
a.push(4);
a.push(2);
a.push(1);
cout << a.min() << endl;
a.pop();
cout << a.min() << endl;
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: