题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
解题思路:
刚开始看到这个题还以为是要自己实现栈的底层数据结构,原来是借助stack栈的数据结构设计算法即可,一般笔试题也不会考写底层实现,倒是面试的时候有可能会考实现底层数据结构,如实现string类,写一个Linklist等。该题的思路是借助一个辅助栈来维护最小的元素的存储。
解题代码:
class Solution {
public:
//栈的实现有两种方式,顺序存储结构,链式存储结构
stack<int> dataStack,minStack;
void push(int value) {
dataStack.push(value);
if(minStack.empty())
minStack.push(value);
else
{
int min=minStack.top();
value<=min ? minStack.push(value):minStack.push(min);
}
}
void pop() {
dataStack.pop();
minStack.pop();
}
int top() {
return dataStack.top();
}
int min() {
//栈的算法,最小元素
return minStack.top();
}
};