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。这样遍历完整个数组即可得到答案。

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;
}