题目:
定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。
解法:
利用两个栈实现,stack1和stack2. stack1实现栈的基本功能pop();top()等,Stack2实现Min()功能,
每当一个元素value压入stack1中时候,比较value 和 stack2.top()元素,把min(value,stack2.top())再压入stack2中,
相当于在stack2中保存每一步的最小元素,在pop()弹出时候,stack1和stack2同步压出即可。
代码如下:
void push(int value) {
stack1.push(value);
if(stack2.empty())
stack2.push(value);
else//栈2不为空
{
if(value<stack2.top())
stack2.push(value);
else
stack2.push(stack2.top());
}
}
void pop() {
stack1.pop();
stack2.pop();
}
int top() {
return stack1.top();
}
int min() {
return stack2.top();
}
stack<int> stack1;
stack<int> stack2;