目录

3101. 交替子数组计数

题目描述：

实现代码与解析：

简洁版：

原理思路：

---

3101. 交替子数组计数

题目描述：

给你一个二进制数组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 。

实现代码与解析：

class Solution {public static void main(String[] args) {int[] nums = new int[]{0,1,1,1};long l = countAlternatingSubarrays(nums);System.out.println(l);}public static long countAlternatingSubarrays(int[] nums) {int pre = -1;int cnt = 0;long res = 0;for (int i = 0; i < nums.length; i++) {if (nums[i] != pre) {cnt++;pre = nums[i];} else {long t = 0;for (int j = 2; j <= cnt; j++) {t += (cnt - j + 1);}cnt = 1;res += t;}}// system.out.println(cnt); 4if (cnt != 1) {long t = 0;for (int j = 2; j <= cnt; j++) {t += (cnt - j + 1);}res += t;}return res + nums.length;}
}

简洁版：

class Solution {public long countAlternatingSubarrays(int[] nums) {long ans = 0;int cnt = 0;for (int i = 0; i < nums.length; i++) {if (i > 0 && nums[i] != nums[i - 1]) {cnt++;} else {cnt = 1;}ans += cnt; }return ans;}
}

原理思路：

        找出cnt与答案的关系即可。我写的是找出每个长度组合个数和cnt的关系，最后加上每一个单个数字。简介版是找出每个已当前为结尾的子组的组合个数。