剑指offer20——包含min函数的栈

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。

题解

#include <iostream>
#include <stack>

using namespace std;

stack<int> s, smin;

void push(int value)
{
	s.push(value);
	if (smin.empty() || value <= smin.top())
		smin.push(value);
}

void pop()
{
	if (s.top() == smin.top())
		smin.pop();
	s.pop();
}

int top()
{
	return s.top();
}

int min()
{
	return smin.top();
}

int main()
{
	ios::sync_with_stdio(false);
	int temp;
	while (1)
	{
		cin >> temp;
		if (!temp)
			break;
		push(temp);
	}
	cout << min() << endl;
	return 0;
}