自学内容网 自学内容网

leetcode-每日一题

3101. 交替子数组计数icon-default.png?t=N7T8https://leetcode.cn/problems/count-alternating-subarrays/

给你一个

二进制数组

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 long countAlternatingSubarrays(int[] nums) {
        int len = nums.length;
        if(len == 1){
            return len;
        } 

        long num = len;
        for(int i = 1;i<len;i++){
            int j = i;
            while(i<len && nums[i] != nums[i-1]){
                i++;
            }
            long k1 = i-j;
            long k2 = 1+k1;
            num += (k1*k2)/2;
        }
        return num;
    }
}


原文地址:https://blog.csdn.net/weixin_61810048/article/details/140234816

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!