3101. 交替子数组计数 Medium

给你一个

二进制数组

nums 。

如果一个

子数组

中 不存在 两个 相邻 元素的值 相同 的情况，我们称这样的子数组为 交替子数组 。

返回数组 nums 中交替子数组的数量。

示例 1：

输入： nums = [0,1,1,1]

输出： 5

解释：

以下子数组是交替子数组：[0] 、[1] 、[1] 、[1] 以及 [0,1] 。

示例 2：

输入： nums = [1,0,1,0]

输出： 10

解释：

数组的每个子数组都是交替子数组。可以统计在内的子数组共有 10 个。

提示：

 ·1 <= nums.length <= 105

 ·nums[i] 不是 0 就是 1 。

题目大意：计算数组中交替子数组的个数。

分析：设dp[i]表示以第i个元素结尾的交替子数组的个数。

（1）由于交替子数组必须是数组中连续的元素，因此当nums[i]!=nums[i-1]时，dp[i]=dp[i-1]+1，当nums[i]==nums[i-1]时，dp[i]=1；

（2）由（1）可知，dp[i]的计算只需要依据dp[i-1]，因此可对dp数组降维，设dp表示以当前所遍历元素的上一个元素结尾的交替子数组的个数。

class Solution {
public:
    long long countAlternatingSubarrays(vector<int>& nums) {
        int N=nums.size();
        long long ans=1,dp=1;
        for(int i=1;i<N;++i){
            dp=1+(nums[i]==nums[i-1]? 0:dp);
            ans+=dp;
        }
        return ans;
    }
};