老样子,题目:496. 下一个更大元素 I - 力扣(LeetCode)
题解:代码随想录
AC代码:
class Solution {public int[] nextGreaterElement(int[] nums1, int[] nums2) {Stack<Integer> stack=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);}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[i]>nums2[stack.peek()]){if(hashMap.containsKey(nums2[stack.peek()])){Integer index=hashMap.get(nums2[stack.peek()]);res[index] = nums2[i];}stack.pop();}stack.push(i);}}return res;}
}
和昨天的每日温度差不多,但是要复杂一点
但是也差不多,题目的意思是让我们找出num1每个元素在num2中相应位置从右边起第一个比它大的值,这就好办了,和昨天的几乎就快一样了,只不过是两个数组而已,因为nums1是nums2的子集,其实我们就将nums2放入栈中维护就行了,按照单调递减栈来维护,如果出现需要进行维护的元素了就查看此刻栈顶的值是不是在nums1中出现,是就可以记录当前nums1的下一个更大数了