目录
2454. 下一个更大元素 IV
题目描述:
实现代码与解析:
单调栈 & 优先级队列
原理思路:
2454. 下一个更大元素 IV
题目描述:
给你一个下标从 0 开始的非负整数数组 nums
。对于 nums
中每一个整数,你必须找到对应元素的 第二大 整数。
如果 nums[j]
满足以下条件,那么我们称它为 nums[i]
的 第二大 整数:
j > i
nums[j] > nums[i]
- 恰好存在 一个
k
满足i < k < j
且nums[k] > nums[i]
。
如果不存在 nums[j]
,那么第二大整数为 -1
。
- 比方说,数组
[1, 2, 4, 3]
中,1
的第二大整数是4
,2
的第二大整数是3
,3
和4
的第二大整数是-1
。
请你返回一个整数数组 answer
,其中 answer[i]
是 nums[i]
的第二大整数。
示例 1:
输入:nums = [2,4,0,9,6] 输出:[9,6,6,-1,-1] 解释: 下标为 0 处:2 的右边,4 是大于 2 的第一个整数,9 是第二个大于 2 的整数。 下标为 1 处:4 的右边,9 是大于 4 的第一个整数,6 是第二个大于 4 的整数。 下标为 2 处:0 的右边,9 是大于 0 的第一个整数,6 是第二个大于 0 的整数。 下标为 3 处:右边不存在大于 9 的整数,所以第二大整数为 -1 。 下标为 4 处:右边不存在大于 6 的整数,所以第二大整数为 -1 。 所以我们返回 [9,6,6,-1,-1] 。
示例 2:
输入:nums = [3,3] 输出:[-1,-1] 解释: 由于每个数右边都没有更大的数,所以我们返回 [-1,-1] 。
提示:
1 <= nums.length <= 105
0 <= nums[i] <= 109
实现代码与解析:
单调栈 & 优先级队列
class Solution {public int[] secondGreaterElement(int[] nums) {Stack<Integer> stk = new Stack<>();PriorityQueue<int[]> pq = new PriorityQueue<>((a, b) -> a[0] - b[0]);int[] res = new int[nums.length];Arrays.fill(res, -1);for (int i = 0; i < nums.length; i++) {while (!pq.isEmpty() && pq.peek()[0] < nums[i]) {int t = pq.peek()[1];pq.poll();res[t] = nums[i];}while (!stk.isEmpty() && nums[stk.peek()] < nums[i]) {int t = stk.peek();stk.pop();pq.offer(new int[]{nums[t], t});}stk.push(i);}return res;}
}
原理思路:
遍历数组,单调栈,求出第一个大于对应位置的数,若有,放入优先级队列中。
所以,在单调栈中的数,代表寻找第一个大于其的数是否存在。
优先级队列里的数,代表寻找第二个大于其的数,只有在单调栈中判断后弹出
的数才会放入优先级队列中。结果数组也是在优先级队列中求出。
优先级队列是为了排序,方便数之间的对比,弹出,然后求出对应位置的结果。