【面试题】剑指Offer-21-包含Min函数的栈

来源:互联网 发布:模拟退火算法c代码 编辑:程序博客网 时间:2024/05/17 01:01

题目概述


这道题目很熟悉了

在栈和队列面试题中遇到并讲解过

解决方法

在栈中多用一个栈来保存最小值

当Min栈为空,或者入栈的元素小于Min栈的栈顶元素,则在入元素栈的同时

将该元素也一并入到Min栈中

从而保证了Min栈的栈顶元素永远都是最小值

出栈时,如果出栈的元素是最小元素,则也弹出Min栈的栈顶元素

否则,仅仅元素出栈即可

代码实现

#include<iostream>using namespace std;#include<stack>#include<assert.h>template<typename T>class StackWithMin{public://特点,存元素的栈和最小元素栈的元素大小是相同的//这里可以优化,如果小的话就存,否则最小元素栈再存//改动后,出栈时判断出栈的元素和最小元素栈的栈顶元素大小关系,如果相等,最小元素栈要出栈void Push(const T& data){s_Elem.push(data);//判断入栈的元素和最小元素栈的栈顶元素的大小关系//来判断最小元素栈入的元素的值if (s_Min.empty() || data < s_Min.top())s_Min.push(data);elses_Min.push(s_Min.top());}void Pop(){assert(!s_Min.empty() && !s_Elem.empty());s_Min.pop();s_Elem.pop();}const T& Min()const{return s_Min.top();}protected:stack<T> s_Elem;stack<T> s_Min;};


1 0
原创粉丝点击