3101.交替子数组计数.java

/*
 * @lc app=leetcode.cn id=3101 lang=java
 *
 * [3101] 交替子数组计数
 *
 * https://leetcode.cn/problems/count-alternating-subarrays/description/
 *
 * algorithms
 * Medium (52.15%)
 * Likes:    11
 * Dislikes: 0
 * Total Accepted:    8.3K
 * Total Submissions: 14.6K
 * Testcase Example:  '[0,1,1,1]'
 *
 * 给你一个二进制数组 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 <= 10^5
 * nums[i] 不是 0 就是 1 。
 * 
 * 
 */

// @lc code=start
class Solution {
    public long countAlternatingSubarrays(int[] nums) {
        int count = 0;
        long result = 0;

        for (int i = 0; i < nums.length; i++) {
            if (i > 0 && nums[i] != nums[i - 1])
                count++;
            else
                count = 1;

            result += count;
        }

        return result;
    }
}
// @lc code=end