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;
}
相关推荐
薰衣草23335 小时前
一天两道力扣(6)
算法·leetcode
逝雪Yuki5 小时前
Leetcode——287. 寻找重复数
c++·leetcode·二分查找·双指针·环形链表
科大饭桶6 小时前
数据结构自学Day13 -- 快速排序--“前后指针法”
数据结构·算法·leetcode·排序算法·c
@蓝莓果粒茶8 小时前
LeetCode第350题_两个数组的交集II
c++·python·学习·算法·leetcode·职场和发展·c#
设计师小聂!8 小时前
力扣热题100----------53最大子数组和
java·数据结构·算法·leetcode
艾莉丝努力练剑10 小时前
【LeetCode&数据结构】二叉树的应用(二)——二叉树的前序遍历问题、二叉树的中序遍历问题、二叉树的后序遍历问题详解
c语言·开发语言·数据结构·学习·算法·leetcode·链表
YuTaoShao10 小时前
【LeetCode 热题 100】51. N 皇后——回溯
java·算法·leetcode·职场和发展
花开富贵ii12 小时前
代码随想录算法训练营二十八天|动态规划part01
java·数据结构·算法·leetcode·动态规划
技术卷12 小时前
详解力扣高频SQL50题之1164. 指定日期的产品价格【中等】
sql·leetcode·oracle
技术卷17 小时前
详解力扣高频SQL50题之610. 判断三角形【简单】
sql·leetcode·oracle