1. 每日温度
维护一个单调递减(可以等于)的栈,栈内存的是元素的下标
当前元素小于等于栈顶元素时,直接入栈
否则,将栈顶元素出栈(对于栈顶元素,它找到了它的答案,即下标之差)
循环比较记录结果,直到当前元素入栈后,栈依然单调递减
class Solution {
public int[] dailyTemperatures(int[] temperatures) {
int length = temperatures.length;
int[] res = new int[length];
Stack<Integer> stack = new Stack<>();
stack.push(0);
for(int i = 1; i < length; i++){
if(temperatures[i] <= temperatures[stack.peek()]){
stack.push(i);
}else{
while(!stack.isEmpty() && temperatures[i] > temperatures[stack.peek()]){
res[stack.peek()] = i - stack.peek();
stack.pop();
}
stack.push(i);
}
}
return res;
}
}
2. 下一个更大元素 I
和前面差不多,只不过需要返回的结果要处理下
class Solution {
public int[] nextGreaterElement(int[] nums1, int[] nums2) {
Map<Integer, Integer> map = new HashMap<>();
for(int i = 0; i < nums1.length; i++){
map.put(nums1[i], i);
}
Stack<Integer> stack = new Stack<>();
stack.push(nums2[0]);
int[] res = new int[nums1.length];
Arrays.fill(res, -1);
for(int i = 0; i < nums2.length; i++){
if(nums2[i] <= stack.peek()){
stack.push(nums2[i]);
}else{
while(!stack.isEmpty() && nums2[i] > stack.peek()){
if(map.containsKey(stack.peek())){ // 只有当栈顶元素需要被记录时才记录
int idx = map.get(stack.peek()); // 取出记录该在的位置
res[idx] = nums2[i];
}
stack.pop(); // 不需要记录的元素也需要出栈,继续看前一个元素
}
stack.push(nums2[i]);
}
}
return res;
}
}