代码随想录单调栈——下一个更大元素 I

题目

给你两个 没有重复元素 的数组 nums1 和 nums2 ，其中nums1 是 nums2 的子集。

请你找出 nums1 中每个元素在 nums2 中的下一个比其大的值。

nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在，对应位置输出 -1
。

示例 1:

输入: nums1 = [4,1,2], nums2 = [1,3,4,2]. 输出: [-1,3,-1] 解释: 对于 num1 中的数字
4 ，你无法在第二个数组中找到下一个更大的数字，因此输出 -1 。 对于 num1 中的数字 1 ，第二个数组中数字1右边的下一个较大数字是
3 。 对于 num1 中的数字 2 ，第二个数组中没有下一个更大的数字，因此输出 -1 。

示例 2: 输入: nums1 = [2,4], nums2 = [1,2,3,4]. 输出: [3,-1] 解释: 对于 num1
中的数字 2 ，第二个数组中的下一个较大数字是 3 。 对于 num1 中的数字 4 ，第二个数组中没有下一个更大的数字，因此输出-1 。

提示：

1 <= nums1.length <= nums2.length <= 1000 0 <= nums1[i], nums2[i] <=
10^4 nums1和nums2中所有整数 互不相同 nums1 中的所有整数同样出现在 nums2 中

思路

每日温度中是求每个元素下一个比当前元素大的元素的位置
本题简单理解就是，num1中选择一个元素，然后在nums2中找到相同的元素，并且在nums2中找到下一个比这个元素大的元素

即最后要求的是nums1的每个元素在nums2中下一个比当前元素大的元素，那么就要定义一个和nums1一样大小的数组result来存放结果，并且初始化为-1

同时题目说是没有重复元素的两个数组，所以可以考虑用map做映射，相当于先在nums2中找满足后一个 > 前一个的元素，然后去hashmap中判断nums2的这个前一个元素在nums1中是否存在，存在的话即满足题意

所以预处理代码为：

HashMap<Integer,Integer> hashMap = new HashMap<>();
for(int i = 1; i < nums1.length; i++){
	hashMap.put(nums[i],i);
}
1
2
3
4

然后使用单调栈，就要确定顺序，维护递增栈还是递减栈

单调栈：

1. 递减栈：栈底到栈顶逐渐递减；如果当前元素比栈顶元素小，直接入栈；如果当前元素比栈顶元素大，则出栈顶元素，将当前元素入栈，保证栈内元素递减；
2. 递增栈：栈底到栈顶逐渐递增；如果当前元素比栈顶元素大，直接入栈；如果当前元素比栈顶元素小，则出栈顶元素，将当前元素入栈，保证栈内元素递增

本题和每日温度一样，都是要找右边（下一个）比自己大的元素，所以是维护递减栈（栈的顺序是从栈底->栈顶），因为只有是递减栈，才能找到右边第一个比自己大的元素，因为只有当前元素比自己大，为了保证递减才会进行操作，出栈顶元素，当前元素进栈，如果是递增栈，遇到比自己大的元素统统进栈，没有操作，就找不到了。

比较栈顶元素和当前元素：

（维护递减栈，栈中存nums2的下标）

1. 当前元素小于栈顶元素，直接进栈
2. 当前元素等于栈顶元素，也是入栈（因为要求的是第一个比自己大的元素，不是大于等于）
3. 当前元素大于栈顶元素，先判断栈顶元素（nums2的下标）是否在nums1中出现过，如果出现过，那么记录结果

java代码如下：

class Solution{
	public int[] nextGreaterElement(int[] nums1, int[] nums2){
		HashMap<Integer,Integer> map = new HashMap<>():
		for(int i = 0 ; i < nums1.length; i++){
			map.put(nums1[i],i);//key:下标元素，value：下标
		}
		
		int[] res = new int[nums1.length];
		Stack<Integer> stack = new Stack<>();
		Arrays.fills(res,-1);
		
		for(int i = 0; i < nums2.length; i++){//维护递减栈，并且栈中存nums2的下标
			while(!stack.isEmpty() && nums2[stack.peek()] < nums2[i]){//如果当前元素大于栈顶元素 
				int pre = nums2[stack.pop()];//取出栈顶元素（下标）在nums2中对应的下标元素
				if(map.containsKey(pre)){//判断下标元素在nums1中是否出现过
					//如果存在的话，则记录结果，res数组中存对应的值，res中的下标是nums1的下标，res中的值是nums2中的值
					res[map.get(pre)] = nums2[i];
				}
			}
			stack.push(i);//while条件不成立的话，直接入栈
		}
		return res;
	}
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24