解决问题类型
The Greater Element
1.模板
如题:给你⼀个数组 [2,1,2,4,3],你返回数组 [4,2,4,-1,-1]。
解释:第⼀个 2 后⾯⽐ 2 ⼤的数是 4; 1 后⾯⽐ 1 ⼤的数是 2;第⼆个 2 后⾯ ⽐ 2 ⼤的数是 4; 4 后⾯没有⽐ 4 ⼤的数,填 -1;3 后⾯没有⽐ 3 ⼤的数,填 -1。
vector<int> nextGreaterElement(vector<int>& nums) {
vector<int> ans(nums.size());
stack<int> s;
for (int i = nums.size()-1; i >= 0; i--) {// 倒着往栈放
while(!s.empty() && s.top() <= nums[i]) {//判定个子高矮
s.pop(); // 矮个子起开,反正被挡住了
}
ans[i] = s.empty() ? -1 : s.top(); // 这个元素身后第一高个子
s.push(nums[i]); // 进队,进行身高判定
}
return ans;
}
2.一些例题
给你⼀个数组 T = [73, 74, 75, 71, 69, 72, 76, 73],这个数组存放的是近⼏天 的天⽓⽓温(这⽓温是铁板烧?不是的,这⾥⽤的华⽒度)。你返回⼀个数 组,计算:对于每⼀天,你还要⾄少等多少天才能等到⼀个更暖和的⽓温; 如果等不到那⼀天,填 0 。
举例:给你 T = [73, 74, 75, 71, 69, 72, 76, 73],你返回 [1, 1, 4, 2, 1, 1, 0, 0]。
解释:第⼀天 73 华⽒度,第⼆天 74 华⽒度,⽐ 73 ⼤,所以对于第⼀天, 只要等⼀天就能等到⼀个更暖和的⽓温。后⾯的同理。
vector<int> dailyTemperatures(vector<int>& T) {
vector<int> ans(T.size());
stack<int> s;// 丢进去的是元素索引喔~
for (int i = T.size()-1; i >= 0; i--) {// 倒着往栈放
while(!s.empty() && T[s.top()] <= T[i]) {// 判断嘛~
s.pop(); // 去掉被挡住的
}
ans[i] = s.empty() ? -1 : (s.top()-i); // 得到索引间距
s.push(i); // 把元素索引丢进去
}
return ans;
}
3.更多练习
Leetcode 42:接雨水
Leetcode 84: ???
Leetcode 496:下一个更大元素 I
Leetcode 503:下一个更大元素 II
Leetcode 739:每日温度
原文参考:
https://github.com/labuladong/fucking-algorithm