刷题-Leetcode-496. 下一个更大元素 I
作者:互联网
496. 下一个更大元素 I
题目链接
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/next-greater-element-i/
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题目描述
题目分析
两个 没有重复元素 的数组 nums1 和 nums2,所以可以用map做映射。
先找nums2的“最大”,找到后再映射(如果nums1中存在)到nums1。
注意:res中找的不是索引,是值。
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
int n1 = nums1.size();
int n2 = nums2.size();
vector<int> res(n1, -1);
if(n1 == 0) return res;
stack<int> st;
st.push(0);
unordered_map<int, int> umap;
for(int i = 0; i < n1; i++){
umap[nums1[i]] = i;
}
for(int i = 1; i < n2; i++){
while(!st.empty() && nums2[i] > nums2[st.top()]){//
if(umap.count(nums2[st.top()]) > 0){//nums2的值有在nums1中的
int index = umap[nums2[st.top()]];
res[index] = nums2[i];
}
st.pop();
}
st.push(i);
}
return res;
}
};
标签:int,res,st,vector,Leetcode,496,nums1,nums2,刷题 来源: https://blog.csdn.net/qq_42771487/article/details/120280897