一、每日一题
给你一个下标从 0 开始的正整数数组 nums
。
如果 nums
的一个子数组满足:移除这个子数组后剩余元素严格递增,那么我们称这个子数组为移除递增子数组。比方说,[5, 3, 4, 6, 7]
中的 [3, 4]
是一个移除递增子数组,因为移除该子数组后,[5, 3, 4, 6, 7]
变为 [5, 6, 7]
,是严格递增的。
请你返回 nums
中移除递增子数组的总数目。
注意 ,剩余元素为空的数组也视为是递增的。
子数组指的是一个数组中一段连续的元素序列。
示例 1:
输入:nums = [1,2,3,4] 输出:10 解释:10 个移除递增子数组分别为:[1], [2], [3], [4], [1,2], [2,3], [3,4], [1,2,3], [2,3,4] 和 [1,2,3,4]。移除任意一个子数组后,剩余元素都是递增的。注意,空数组不是移除递增子数组。
示例 2:
输入:nums = [6,5,7,8] 输出:7 解释:7 个移除递增子数组分别为:[5], [6], [5,7], [6,5], [5,7,8], [6,5,7] 和 [6,5,7,8] 。 nums 中只有这 7 个移除递增子数组。
示例 3:
输入:nums = [8,7,6,6] 输出:3 解释:3 个移除递增子数组分别为:[8,7,6], [7,6,6] 和 [8,7,6,6] 。注意 [8,7] 不是移除递增子数组因为移除 [8,7] 后 nums 变为 [6,6] ,它不是严格递增的。
提示:
1 <= nums.length <= 105
1 <= nums[i] <= 109
二、代码
class Solution:
def incremovableSubarrayCount(self, a: List[int]) -> int:
n = len(a)
i = 0
while i < n - 1 and a[i] < a[i + 1]:
i +=1
if i == n - 1:
return n * (n + 1) // 2
ans = i + 2
j = n - 1
while j == n - 1 or a[j] < a[j + 1]:
while i >= 0 and a[i] >= a[j]:
i -= 1
ans += i + 2
j -= 1
return ans