经典的滑动窗口的题目 力扣 2799. 统计完全子数组的数目(面试题)

时间:2024-06-03 12:50:46

给你一个由  整数组成的数组 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 。

使用滑动窗口来解决 ,定义一个a[20000],数组的大小可以参考具体数据的范围。

重点是理解ans=len(nums)-right.举一个简单的例子

代码如下:

ans=0
a=[0]*20000
left=0
norep=len(set(nums))
m=0#定义出现新元素的次数
for right in range(len(nums)):
    a[nums[right]]+=1
    if a[nums[right]]==1:
        m+=1
    while m==norep:
        ans+=len(nums)-right
        a[nums[left]]-=1
        if a[nums[left]]==0:#说明消失了一个新元素,就不满足条件了
            m-=1
        left+=1
print(ans)

 

  1. ans=0:初始化一个变量ans,用于存储最终结果,即满足条件的子数组的总数。

  2. a=[0]*20000:初始化一个长度为20000的数组a,用于记录每个数字在当前窗口中出现的次数。

  3. left=0:初始化左指针left,用于表示当前窗口的左边界。

  4. norep=len(set(nums)):通过set(nums)来获得数组nums中不重复元素的个数,并将其赋值给norep,表示不重复元素的数量。

  5. m=0:初始化变量m,用于记录当前窗口中出现的新元素的次数。

  6. for right in range(len(nums))::遍历数组nums中的每个元素,right表示当前窗口的右边界。

  7. a[nums[right]]+=1:将当前元素nums[right]在数组a中的计数加1。

  8. if a[nums[right]]==1::如果当前元素nums[right]在当前窗口中第一次出现,则将新元素的数量m加1。

  9. while m==norep::当当前窗口中的新元素数量等于不重复元素的总数时,执行下面的操作。

  10. ans+=len(nums)-right:将当前满足条件的子数组的数量累加到结果ans中。这里使用len(nums)-right来计算当前窗口中满足条件的子数组的个数。

  11. a[nums[left]]-=1:将左边界元素在数组a中的计数减1。

  12. if a[nums[left]]==0::如果左边界元素在当前窗口中消失,即计数为0,则将m减1,表示新元素数量减少了一个。

  13. left+=1:将左指针向右移动,缩小窗口。

  14. 最后输出ans,即满足条件的子数组的总数。

 

主要目的是计算数组中所有满足特定条件的子数组的总数。具体来说,它通过维护一个滑动窗口,不断调整窗口的左右边界,来统计满足条件的子数组的数量。

该条件是:子数组中包含的不同元素的数量等于整个数组中不同元素的数量。

通过遍历数组,并在遍历过程中维护窗口,当窗口中包含的不同元素数量等于整个数组中不同元素的数量时,就计算当前窗口中满足条件的子数组的数量,并累加到结果中。最终,输出结果即为满足条件的子数组的总数。

这种算法的思想是利用滑动窗口来遍历所有可能的子数组,并在满足条件时进行统计,以提高效率。