给你一个二进制数组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
。
java
class Solution {
public long countAlternatingSubarrays(int[] nums) {
long ans = 1;
for(int end = 1,start = 0;end <nums.length;end++){
if(nums[end] == nums[end -1]){
start = end;
}
//如果一个数组为交替
ans += end - start + 1; //end-start+1计算的是start到end的子数组数量,但是单个只包含end
}
return ans;
}
}