LeetCode-2454. 下一个更大元素 IV【栈 数组 二分查找 排序 单调栈 堆(优先队列)】

LeetCode-2454. 下一个更大元素 IV【栈 数组 二分查找 排序 单调栈 堆(优先队列)】

  • 题目描述:
  • 解题思路一:单调栈,典型的单调栈问题,不过这里用的是双单调栈,因为是找第二大整数。我们用第一个栈来维护下一个更大整数,然后用第二个栈来得出答案(即第二大整数)。还是遇到当前元素大于栈顶元素就出栈,出栈时维护答案。
  • 解题思路二:0
  • 解题思路三:0

题目描述:

给你一个下标从 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


你可能感兴趣的:(算法题,leetcode,算法,职场和发展,单调栈,数组)