赞
踩
给定一个整数数组 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;
- }
- };
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;
- }
- };
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。