文章目錄

• • 1. 題目
  • 2. 解題
  • • 2.1 雙棧解法
    • 2.2 list+map

1. 題目

設計一個最大棧，支持 push、pop、top、peekMax 和 popMax 操作。

push(x) -- 將元素 x 壓入棧中。 pop() -- 移除棧頂元素并返回這個值。 top() -- 返回棧頂元素。 peekMax() -- 返回棧中最大元素。 popMax() -- 返回棧中最大的元素，并將其刪除。如果有多個最大元素，只要刪除最靠近棧頂的那個。樣例 1: MaxStack stack = new MaxStack(); stack.push(5); stack.push(1); stack.push(5); stack.top(); -> 5 stack.popMax(); -> 5 stack.top(); -> 1 stack.peekMax(); -> 5 stack.pop(); -> 1 stack.top(); -> 5注釋: -1e7 <= x <= 1e7 操作次數不會超過 10000。 當棧為空的時候不會出現后四個操作。

來源：力扣（LeetCode） 鏈接：https://leetcode-cn.com/problems/max-stack
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

類似題目：LeetCode 155. 最小棧

2.1 雙棧解法

• 同時插入數值，和最大值
• 當要刪除最大的值的時候，要將不是最大值的數，先存入臨時棧，后序再挪回來，最壞時間復雜度O(n)

class MaxStack {int maxelem = INT_MIN;stack<int> s;stack<int> temp;int v, m; public:/** initialize your data structure here. */MaxStack() {}void push(int x) {maxelem = max(maxelem, x);s.push(x);s.push(maxelem);}int pop() {s.pop();v = s.top();s.pop();maxelem = s.empty() ? INT_MIN : s.top();return v;}int top() {m = s.top();s.pop();v = s.top();s.push(m);return v;}int peekMax() {return s.top();}int popMax() {int ans = s.top();maxelem = s.top();bool flag = true;while(flag){s.pop();if(s.top() != maxelem)temp.push(s.top());elseflag = false;s.pop();}maxelem = s.empty() ? INT_MIN : s.top();while(!temp.empty()){v = temp.top();temp.pop();s.push(v);maxelem = max(maxelem, v);s.push(maxelem);}return ans;} };

140 ms 32.2 MB

2.2 list+map

• list 當做棧來使用
• map的key為數值，value掛著數值下，對應的list迭代器
• 時間復雜度O(log n)

class MaxStack {list<int> l;map<int, vector<list<int>::iterator>> m; public:/** initialize your data structure here. */MaxStack() {}void push(int x) {l.push_front(x);m[x].push_back(l.begin());}int pop() {int v = l.front();m[v].pop_back();if(m[v].empty())m.erase(v);l.pop_front();return v;}int top() {return l.front();}int peekMax() {return m.rbegin()->first;}int popMax() {int v = m.rbegin()->first;auto it = m[v].back();m[v].pop_back();if(m[v].empty())m.erase(v);l.erase(it);return v;} };

240 ms 35.2 MB

---

長按或掃碼關注我的公眾號，一起加油、一起學習進步！

總結

以上是生活随笔為你收集整理的LeetCode 716. 最大栈（双栈 / list+map）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。