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] 是如上所述的 下一个更大元素 。
示例 1:
输入:nums1 = [4,1,2], nums2 = [1,3,4,2].
输出:[-1,3,-1]
解释:nums1 中每个值的下一个更大元素如下所述:
- 4 ,用加粗斜体标识,nums2 = [1,3,4,2]。不存在下一个更大元素,所以答案是 -1 。
- 1 ,用加粗斜体标识,nums2 = [1,3,4,2]。下一个更大元素是 3 。
- 2 ,用加粗斜体标识,nums2 = [1,3,4,2]。不存在下一个更大元素,所以答案是 -1 。
nums2 = [1,3,4,2],倒着入栈,先入2;
到4时,发现4比2大,出掉栈中比4小的元素,4入栈;
到3时,发现3比4小,那么栈顶的元素4就是它右侧第一个比它大的元素,3入栈;
到1时,发现1比3小,那么栈顶的元素3就是它右侧第一个比它大的元素,1入栈。
观上图也可见,栈中存的就是在你右侧且比你还大的元素,栈顶就是第一个比你大的元素。
分析它的时间复杂度:总共有 n
个元素,每个元素都被 push
入栈了一次,而最多会被 pop
一次,没有任何冗余操作。所以总的计算规模是和元素规模 n
成正比的,也就是 O(n)
的复杂度。
- class Solution {
- public:
- vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
- unordered_map<int,int> tmp;
- greaterStack(tmp,nums2);
- vector<int> res;
- for(int i=0;i
size();i++) - {
- res.push_back(tmp[nums1[i]]);
- }
- return res;
- }
- void greaterStack(unordered_map<int,int>& tmp,vector<int>& nums2)
- {
- stack<int> s;
- int n=nums2.size();
- for(int i=n-1;i>=0;i--)//倒着入栈,正着出栈
- {
- //出尽栈中比nums[i]还小的,之后栈顶元素就是nums[i]右侧第一个比它大的元素了
- while(!s.empty()&&nums2[i]>=s.top())
- {
- s.pop();
- }
- if(s.empty())//为空代表右侧没有比它大的元素
- {
- tmp[nums2[i]]=-1;
- }
- else//栈顶元素就是nums[i]右侧第一个比它大的元素
- {
- tmp[nums2[i]]=s.top();
- }
- s.push(nums2[i]);
- }
- }
- };
给定一个循环数组 nums ( nums[nums.length - 1] 的下一个元素是 nums[0] ),返回 nums 中每个元素的 下一个更大元素 。
数字 x 的 下一个更大的元素 是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1 。
示例 1:
输入: nums = [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2;
数字 2 找不到下一个更大的数;
第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
示例 2:
输入: nums = [1,2,3,4,3]
输出: [2,3,4,-1,4]
将数组长度翻倍模拟环形数组
- class Solution {
- public:
- vector<int> nextGreaterElements(vector<int>& nums)
- {
- vector<int> res(nums.size());
- stack<int> s;
- int n=nums.size();
- for(int i=2*n-1;i>=0;i--)//数组长度加倍模拟环形数组
- {
- while(!s.empty()&&nums[i%n]>=s.top())
- {
- s.pop();
- }
- if(s.empty())
- {
- res[i%n]=-1;
- }
- else
- {
- res[i%n]=s.top();
- }
- s.push(nums[i%n]);
- }
- return res;
- }
- };
类似题目:
739. 每日温度 - 力扣(LeetCode) (通过用栈存储元素下标的方式减小空间消耗)