LeetCode 2762.不间断子数组

给你一个下标从 0 开始的整数数组 nums 。nums 的一个子数组如果满足以下条件,那么它是 不间断 的:

i,i + 1 ,…,j 表示子数组中的下标。对于所有满足 i <= i1, i2 <= j 的下标对,都有 0 <= |nums[i1] - nums[i2]| <= 2 。
请你返回 不间断 子数组的总数目。

子数组是一个数组中一段连续 非空 的元素序列。

示例 1:

输入:nums = [5,4,2,4]
输出:8
解释:
大小为 1 的不间断子数组:[5], [4], [2], [4] 。
大小为 2 的不间断子数组:[5,4], [4,2], [2,4] 。
大小为 3 的不间断子数组:[4,2,4] 。
没有大小为 4 的不间断子数组。
不间断子数组的总数目为 4 + 3 + 1 = 8 。
除了这些以外,没有别的不间断子数组。
示例 2:

输入:nums = [1,2,3]
输出:6
解释:
大小为 1 的不间断子数组:[1], [2], [3] 。
大小为 2 的不间断子数组:[1,2], [2,3] 。
大小为 3 的不间断子数组:[1,2,3] 。
不间断子数组的总数目为 3 + 2 + 1 = 6 。

提示:

1 <= nums.length <= 10 5 ^5 5
1 <= nums[i] <= 10 9 ^9 9

滑动窗口,窗口是不间断子数组时,窗口的任一子数组也是不间断子数组:

class Solution {
public:
    long long continuousSubarrays(vector<int>& nums) {
        long long ans = 0;
        int left = 0;
        map<int, int> cnt;
        for (int i = 0; i < nums.size(); ++i) {
            ++cnt[nums[i]];

            while (cnt.rbegin()->first - cnt.begin()->first > 2) {
                if (--cnt[nums[left]] == 0) {
                    cnt.erase(nums[left]);
                }

                ++left;
            }

            ans += i - left + 1;
        }
        return ans;
    }
};

如果nums的大小为n,则此算法时间复杂度为O(n),空间复杂度为O(1),cnt中最多只有3个数。

你可能感兴趣的:(LeetCode,leetcode,算法,数据结构)