一、每日温度
题目一:739. 每日温度
给定一个整数数组 temperatures
,表示每天的温度,返回一个数组 answer
,其中 answer[i]
是指对于第 i
天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用 0
来代替
如果明天的温度比今天高,那么栈顶的元素就可以出栈。栈内保存的是索引,栈顶的元素是当前未被高温度覆盖的最近的一天。
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
int n = temperatures.size();
vector<int> answer(n);
stack<int> s;
for(int i = 0; i < n; ++i){
while(!s.empty() && temperatures[s.top()] < temperatures[i]){
int preIndex = s.top();
answer[preIndex] = i - preIndex;
s.pop();
}
s.push(i);
}
while(!s.empty()){
answer[s.top()] = 0;
s.pop();
}
return answer;
}
};
二、下一个更大元素 I
题目一:496. 下一个更大元素 I
nums1
中数字 x
的 下一个更大元素 是指 x
在 nums2
中对应位置 右侧 的 第一个 比 x
大的元素。
给你两个 没有重复元素 的数组 nums1
和 nums2
,下标从 0 开始计数,其中nums1
是 nums2
的子集。
对于每个 0 <= i < nums1.length
,找出满足 nums1[i] == nums2[j]
的下标 j
,并且在 nums2
确定 nums2[j]
的 下一个更大元素 。如果不存在下一个更大元素,那么本次查询的答案是 -1
。
返回一个长度为 nums1.length
的数组 ans
作为答案,满足 ans[i]
是如上所述的 下一个更大元素 。
创建一个映射用于存储nums2中每个数的下一个更大元素,然后再用创建的映射查找nums1中元素的下一个更大元素。
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
unordered_map<int, int> numToNextGreater;
stack<int> nextGreaterStack;
for(int num : nums2){
while(!nextGreaterStack.empty() && nextGreaterStack.top() < num){
numToNextGreater[nextGreaterStack.top()] = num;
nextGreaterStack.pop();
}
nextGreaterStack.push(num);
}
while(!nextGreaterStack.empty()){
numToNextGreater[nextGreaterStack.top()] = -1;
nextGreaterStack.pop();
}
vector<int> answer(nums1.size());
for(int i = 0; i < nums1.size(); ++i){
answer[i] = numToNextGreater[nums1[i]];
}
return answer;
}
};