class Solution {
public:
long long incremovableSubarrayCount(vector<int>& nums) {
}
};
10033. 统计移除递增子数组的数目 II
给你一个下标从 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 <= 1051 <= nums[i] <= 109原站题解
javascript 解法, 执行用时: 87 ms, 内存消耗: 61.6 MB, 提交时间: 2024-07-11 09:10:11
/**
* @param {number[]} nums
* @return {number}
*/
var incremovableSubarrayCount = function(a) {
const n = a.length;
let i = 0;
while (i < n - 1 && a[i] < a[i + 1]) {
i++;
}
if (i === n - 1) { // 每个非空子数组都可以移除
return n * (n + 1) / 2;
}
let ans = i + 2; // 不保留后缀的情况，一共 i+2 个
// 枚举保留的后缀为 a[j:]
for (let j = n - 1; j === n - 1 || a[j] < a[j + 1]; j--) {
while (i >= 0 && a[i] >= a[j]) {
i--;
}
// 可以保留前缀 a[:i+1], a[:i], ..., a[:0] 一共 i+2 个
ans += i + 2;
}
return ans;
};
rust 解法, 执行用时: 15 ms, 内存消耗: 3.7 MB, 提交时间: 2024-07-11 09:09:59
impl Solution {
pub fn incremovable_subarray_count(a: Vec<i32>) -> i64 {
let n = a.len();
let mut i = 0;
while i < n - 1 && a[i] < a[i + 1] {
i += 1;
}
if i == n - 1 { // 每个非空子数组都可以移除
return n as i64 * (n + 1) as i64 / 2;
}
let mut i = i as i64;
let mut ans = i + 2; // 不保留后缀的情况，一共 i+2 个
// 枚举保留的后缀为 a[j:]
let mut j = n - 1;
while j == n - 1 || a[j] < a[j + 1] {
while i >= 0 && a[i as usize] >= a[j] {
i -= 1;
}
// 可以保留前缀 a[:i+1], a[:i], ..., a[:0] 一共 i+2 个
ans += i + 2;
j -= 1;
}
ans
}
}
golang 解法, 执行用时: 104 ms, 内存消耗: 7.8 MB, 提交时间: 2023-12-25 00:37:05
func incremovableSubarrayCount(a []int) int64 {
n := len(a)
i := 0
for i < n-1 && a[i] < a[i+1] {
i++
}
if i == n-1 { // 每个非空子数组都可以移除
return int64(n) * int64(n+1) / 2
}
ans := int64(i + 2)
for j := n - 1; j == n-1 || a[j] < a[j+1]; j-- {
for i >= 0 && a[i] >= a[j] {
i--
}
ans += int64(i + 2)
}
return ans
}
java 解法, 执行用时: 1 ms, 内存消耗: 56.7 MB, 提交时间: 2023-12-25 00:36:54
class Solution {
public long incremovableSubarrayCount(int[] a) {
int n = a.length;
int i = 0;
while (i < n - 1 && a[i] < a[i + 1]) {
i++;
}
if (i == n - 1) { // 每个非空子数组都可以移除
return (long) n * (n + 1) / 2;
}
long ans = i + 2;
for (int j = n - 1; j == n - 1 || a[j] < a[j + 1]; j--) {
while (i >= 0 && a[i] >= a[j]) {
i--;
}
ans += i + 2;
}
return ans;
}
}
cpp 解法, 执行用时: 136 ms, 内存消耗: 92.5 MB, 提交时间: 2023-12-25 00:36:41
class Solution {
public:
long long incremovableSubarrayCount(vector<int> &a) {
int n = a.size();
int i = 0;
while (i < n - 1 && a[i] < a[i + 1]) {
i++;
}
if (i == n - 1) { // 每个非空子数组都可以移除
return (long long) n * (n + 1) / 2;
}
long long ans = i + 2;
for (int j = n - 1; j == n - 1 || a[j] < a[j + 1]; j--) {
while (i >= 0 && a[i] >= a[j]) {
i--;
}
ans += i + 2;
}
return ans;
}
};
python3 解法, 执行用时: 100 ms, 内存消耗: 31.8 MB, 提交时间: 2023-12-25 00:36:30
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