代码随想录
739. 每日温度
思路
思路:从栈头到栈底(每次从栈头压入)升序的单调栈,一旦遇到更大的就先弹出再压入,否则直接压入。也就是每次必然会压入的。
代码
class Solution {
public int[] dailyTemperatures(int[] temperatures) {
int answer[] = new int[temperatures.length];
Stack<Integer> s = new Stack<Integer>();
s.push(0);
for(int i = 1;i<temperatures.length;i++){
while(!s.isEmpty() && temperatures[i]>temperatures[s.peek()]){
int t = s.pop();
answer[t] = i-t;
}
s.push(i);
}
return answer;
}
}
496.下一个更大元素 I
思路
栈头到栈底从小到大
对nums2的元素来进行单调栈的操作,即记录nums2的下标。
但nums2的元素在nums1中出现过的时候,把该索引的位置的元素设置为当前出栈的nums2的元素,即为nums1在nums2中的下一个更大的元素
因为题目强调元素不重复,可以对nums1的元素用map记录,key为元素值,value为索引值,方便用nums2的元素查找1中的索引位置。
代码
class Solution {
public int[] nextGreaterElement(int[] nums1, int[] nums2) {
Stack<Integer> temp = new Stack<>();
int[] res = new int[nums1.length];
Arrays.fill(res,-1);
HashMap<Integer, Integer> hashMap = new HashMap<>();
for (int i = 0 ; i< nums1.length ; i++){
hashMap.put(nums1[i],i);
}
temp.push(0);
for (int i = 1; i < nums2.length; i++) {
while (!temp.isEmpty() && nums2[temp.peek()] < nums2[i]) {
if (hashMap.containsKey(nums2[temp.peek()])){
Integer index = hashMap.get(nums2[temp.peek()]);
res[index] = nums2[i];
}
temp.pop();
}
temp.push(i);
}
return res;
}
}