LeetCode-2454. 下一个更大元素 IV【栈 数组 二分查找 排序 单调栈 堆(优先队列)】
2023-12-14 11:49:38
LeetCode-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:
def secondGreaterElement(self, nums: List[int]) -> List[int]:
stack_1 = []
stack_2 = []
n = len(nums)
ans = [-1] * n
tmp = [] # 暂时存储stack_1的出栈结果
for i in range(n):
while stack_1 and nums[i] > nums[stack_1[-1]]:
num = stack_1.pop()
tmp.append(num)
while stack_2 and nums[i] > nums[stack_2[-1]]:
ans[stack_2[-1]] = nums[i]
stack_2.pop()
stack_1.append(i)
while tmp:
stack_2.append(tmp.pop())
return ans
时间复杂度:O(n) 遍历一次,最多入栈出栈一次
空间复杂度:O(n) 栈
解题思路二:0
解题思路三:0
文章来源:https://blog.csdn.net/qq_45934285/article/details/134989211
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!