刷题神器
往期回顾
>【回溯算法】|代码随想录算法训练营第22天|491.递增子序列、 46.全排列、47.全排列 II、【总结】
题目
739. 每日温度
-
学后思路
使用单调栈解决,注意单调栈里面的存放内容为下标值
解法1:
class Solution {
public int[] dailyTemperatures(int[] temperatures) {
int len = temperatures.length;
int[] res = new int[len];
Deque<Integer> stack = new LinkedList<>();
stack.push(0);
for (int i = 1; i < len; 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;
}
}
- 题目总结
- 注意单调递增还是递减
- 注意栈的逻辑判断条件
496.下一个更大元素 I
- 学后思路
解题思路还是使用单调栈,但是要注意两个数组如何映射
解法一:
class Solution {
public int[] nextGreaterElement(int[] nums1, int[] nums2) {
Stack<Integer> stack = new Stack<>();
// 结果集是nums1的长度
int[] res = new int[nums1.length];
Arrays.fill(res, -1);
// 保存nums1的 value index的映射过线
HashMap<Integer, Integer> hashMap = new HashMap<>();
for (int i = 0; i < nums1.length; i++) {
hashMap.put(nums1[i], i);
}
stack.add(0);
for (int i = 1; i < nums2.length; i++) {
if (nums2[i] <= nums2[stack.peek()]) {
stack.add(i);
} else {
while (!stack.isEmpty() && nums2[stack.peek()] < nums2[i]) {
if (hashMap.containsKey(nums2[stack.peek()])) {
// 判断 nums2中的值是否是nums1中的值 转换index
Integer index = hashMap.get(nums2[stack.peek()]);
res[index] = nums2[i];
}
stack.pop();
}
stack.add(i);
}
}
return res;
}
}
- 题目总结
- 注意result的长度
- 注意nums1与nums2的index转换
- hashMap判断nums2中的值是否在nums1中
503.下一个更大元素II
- 学后思路
循环数组,使用取模的方式进行循环
解法一:
class Solution {
public int[] nextGreaterElements(int[] nums) {
//边界判断
if(nums == null || nums.length <= 1) {
return new int[]{-1};
}
int size = nums.length;
int[] result = new int[size];//存放结果
Arrays.fill(result,-1);//默认全部初始化为-1
Stack<Integer> st= new Stack<>();//栈中存放的是nums中的元素下标
for(int i = 0; i < 2*size; i++) {
while(!st.empty() && nums[i % size] > nums[st.peek()]) {
result[st.peek()] = nums[i % size];//更新result
st.pop();//弹出栈顶
}
st.push(i % size);
}
return result;
}
}
- 题目总结
- 注意循环是2倍的数组长度
- 循环过程中不会发生覆盖的情况,比如234 或者 432,234产生的结果是一样的,432,后面栈中的结果没有处理,所以没有影响