leetcode 2799. 统计完全子数组的数目中等

给你一个由正整数组成的数组 nums 。

如果数组中的某个子数组满足下述条件，则称之为 完全子数组 ：

子数组中不同元素的数目等于整个数组不同元素的数目。

返回数组中 完全子数组 的数目。

子数组 是数组中的一个连续非空序列。

示例 1：

复制代码

输入：nums = [1,3,1,2,2]
输出：4
解释：完全子数组有：[1,3,1,2]、[1,3,1,2,2]、[3,1,2] 和 [3,1,2,2] 。

示例 2：

复制代码

输入：nums = [5,5,5,5]
输出：10
解释：数组仅由整数 5 组成，所以任意子数组都满足完全子数组的条件。子数组的总数为 10 。

提示：

1 <= nums.length <= 1000
1 <= nums[i] <= 2000

分析：使用滑动窗口来解决这个问题。用一个 flag 数组，记录对应的元素是否出现过，并用 cnt 记录出现过不同的元素数量。之后遍历数组，用 temp 数组记录已经出现过哪些元素，每出现过一个元素， cnt 的值减 1。

初始时，窗口的左端点 L = 0。当 cnt 等于 0 时，说明从 L 到当前位置 index 的子数组是完全子数组，且从 index 到 nums 数组末尾的所有子数组也都满足完全子数组的条件，因此 ans += numsSize - index。之后要将窗口的左端点 L 向右移动，直到从 L 到 index 的子数组不满足条件。每次移动时，若仍然满足条件，ans 还要加上 numsSize - index。这样遍历完整个数组即可得到答案。

cpp 复制代码

int countCompleteSubarrays(int* nums, int numsSize) {
    int ans,l,r,cnt;cnt=ans=l=r=0;
    int flag[2020]={0},temp[2020]={0};
    for(int i=0;i<numsSize;++i)
        if(!flag[nums[i]])flag[nums[i]]++,cnt++;

    for(int i=0;i<numsSize;++i)
    {
        temp[nums[i]]++;
        if(temp[nums[i]]==flag[nums[i]])cnt--;
        
        if(cnt==0)
        {
            ans+=numsSize-i;
            //printf("ans=%d i=%d add=%d\n",ans,i,numsSize-i);
            for(;l<=i;++l)
            {
                temp[nums[l]]--;
                if(temp[nums[l]]<flag[nums[l]])
                {
                    l++;cnt++;break;
                }
                else ans+=numsSize-i;
                //printf("for ans=%d l=%d \n",ans,l);
            }
            //printf("ans=%d l=%d \n",ans,l);
        }
    }

    return ans;
}

leetcode 2799. 统计完全子数组的数目 中等

leetcode 2799. 统计完全子数组的数目中等