(本博客旨在个人总结回顾)
题目描述:
定义栈的数据结构,请在改类型中实现一个能够得到栈的最小元素的min函数。在改栈中,调用min、push及pop的时间复杂度都是O(1)。
解题思路:
当题目思路不清晰时可使用画图或表格来模拟步骤,并慢慢发现其中的规律。
步骤 | 操作 | 数据栈 | 辅助栈 | 最小值 |
1 | 压入3 | 3 | 3, | 3 |
2 | 压入4 | 3,4 | 3,3 | 3 |
3 | 压入2 | 3,4,2 | 3,3,2 | 2 |
4 | 压入1 | 3,4,2,1 | 3,3,2,1 | 1 |
5 | 弹出 | 3,4,2 | 3,3,2, | 2 |
6 | 弹出 | 3,4 | 3,3 | 3 |
7 | 压入0 | 3,4,0 | 3,3,0 | 0 |
源码:
// ConsoleApplication1.cpp : 定义控制台应用程序的入口点。
//
#include "stdafx.h"
#include <iostream>
#include <stack>
using namespace std;
template <typename T>
struct StackWithMin
{
void push(const T& value);
void pop(const T& value);
const T min();
stack<T> m_data;
stack<T> m_min;
};
template <typename T>
void StackWithMin<T>::push(const T& 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());
}
}
template <typename T>
void StackWithMin<T>::pop(const T& value)
{
assert(m_data.size() > 0 && m_min.size() > 0);
m_data.pop();
m_min.pop();
}
template <typename T>
const T StackWithMin<T>::min()
{
assert(m_data.size() > 0 && m_min.size() > 0);
return m_min.top();
}
int _tmain(int argc, _TCHAR* argv[])
{
system("pause");
return 0;
}