W:
对于快速查找元素是否在另一个集合中是否出现,条件反射想到哈希表,这里由于需要记录下标,需要使用map;
单调栈的一个特点是他必定是有序的,此题可分解 为两步,一是建立单调栈,栈中存放的是nums2对应的下标(note)
for(int i=1;i<len2;i++){
while(!st.empty() && nums2[i]> nums2[st.top()]){
st.pop();
}
st.push(i);
}
二是进行判断,如果有个元素比栈顶的大,那么判断栈顶元素所对应在nums2的值是否在nums1中出现,由于是按照顺序推入,所以必定
是第一个出现的比nums[i]大的值,那么有些在nums1中的元素没有被判断怎么办,可以肯定的是,如果没有被判断,那么就是没有,以示例1为例:
当栈顶元素为4时,按照顺序;下一个元素为2,同时也是最后一个元素,但2比4小;不能够进入while循环,也就是该位置为原来的值-1
if(umap.count(nums2[st.top()])){
int index=umap[nums2[st.top()]];
res[index]=nums2[i];
}
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
int len1=nums1.size(),len2=nums2.size();
vector<int> res(len1,-1);
if(len1==0) return res;
unordered_map<int,int> umap;// key:下标元素,value:下标
stack<int> st;
for(int i=0;i<len1;i++){
umap[nums1[i]]=i;
}
st.push(0);
for(int i=1;i<len2;i++){
while(!st.empty() && nums2[i]> nums2[st.top()]){
if(umap.count(nums2[st.top()])){
int index=umap[nums2[st.top()]];
res[index]=nums2[i];
}
st.pop();//note pop()
}
st.push(i);
}
return res;
}
};