1. 题目
给你一个下标从 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]
的第二大整数。
2. 示例
示例 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]
3. 思路
这道题是一道关于栈的题目,但他和普通的栈有点不同,我们这里需要使用的是单调栈。所谓单调栈,它是一种从栈顶往栈底单调递增或单调递减的栈。根据题意,他让我们求每个数右边第二个比他大的数,因此我们可以先定义两个列表用于模拟栈的操作,分别为first和second:
对于first栈的规则:如果first栈中的栈顶元素大于当前元素,则当前元素直接进入first栈,如果大于first栈顶元素,则将first栈元素弹出并加入到second栈中,直到first栈顶元素是从下往上单调递减的。
对于second栈的规则:second栈用于存储从first弹出的元素,可以这么理解,first的栈顶元素就是second栈的右边第一个比他大的元素,如果此时来一个比second栈顶大的元素,那么必然是它右边第二个比它大的元素。
下一个最大数IV
4. 代码
class Solution:
def secondGreaterElement(self, nums: List[int]) -> List[int]:
length = len(nums)
ans = [-1 for _ in range(length)] # 用于保存最终的结果
first ,second = [], [] # 定义两个栈
# 遍历数组
for i,x in enumerate(nums):
# 如果second栈不为空同时当前元素x比second栈顶元素大,则x是该元素右边第个比它二大的
while second and nums[second[-1]] < x:
ans[second.pop()] = x
# 临时保存first栈的最后一个索引,方便后面删除
j = len(first) - 1
# 如果j>=0 同时 first栈顶元素小于x,这里是为了将first栈更新为新的单调栈,即栈顶必定是first栈内最小的
while j >= 0 and nums[first[j]] < x:
j -= 1
# first栈弹出的元素都加入到second栈
second += first[j+1:]
del first[j+1:] # 弹出刚才的那些元素(比x小的first元素)
first.append(i) # 将当前元素x索引加入first栈
return ans