写力扣遇见的题,思想方法很好,记录一下。
题目:
给定一个正整数数组 nums和一个整数 k ,返回 num 中 「好子数组」 的数目。
如果 nums 的某个子数组中不同整数的个数恰好为 k,则称 nums 的这个连续、不一定不同的子数组为 「好子数组 」。
例如,[1,2,3,1,2] 中有 3 个不同的整数:1,2,以及 3。
子数组 是数组的 连续 部分。
示例 1:
输入:nums = [1,2,1,2,3], k = 2
输出:7
解释:恰好由 2 个不同整数组成的子数组:[1,2], [2,1], [1,2], [2,3], [1,2,1], [2,1,2], [1,2,1,2].
1 <= nums.length <= 2 *
1 <= nums[i], k <= nums.length
这个题目有几个问题需要解决:
怎么知道当前子数组有多少不同整数?
怎么把那些很细的子数组也给算到我的总和里?
我们用一个变量cnt来记录当前子数组有多少不同整数,用双指针来进行遍历。如果nums[right]没有出现过,cnt++,如果nums[left]出现的此数为1且left将要移动,cnt--。
那怎么知道这个数有没有出现过?只能用数组来记录了(只是单纯因为我不会哈希表)。
这时候我们想,如果cnt刚好等于k,左右指针都停下来的时候,我们要怎么计算这一段区间的子数组呢?
如果你擅长找规律,或者排列组合学的很好,你应该发现了,子数组的个数等于(length)为区间长度。这时候你欣喜若狂,只要我不断移动左右指针,找到所有满足条件的区间,套个公式不就ok了!
但是!这时候就出现问题了,因为区间是可能重合的,那子数组也可能重算了。这时候你心态大崩,开始摆烂(没错就是我了)。
这时候就应该想(偷)一些其他思路了。
这道题最关键的突破就是子数组中不同整数的个数恰好为 k的数目=子数组中不同整数的个数不多于 k的数目-子数组中不同整数的个数不多于 k-1的数目
为什么这样想呢?(我猜的)主要是因为不同整数不多于k比刚好为k的子数目个数好求。
我们要求由最多 K 个不同整数组成的子数组的个数,那么对于长度 [left, right] 区间内的每个子数组都是满足要求的,res 要累加的其实是 符合条件的并且以 right 为右端点的子数组个数,这个数量就是right - left + 1,即区间长度。例如 [2,4,3,5] 满足条件,要累加的其实就是 [2,4,3,5], [4,3,5], [3,5], [5] 四个子区间。
对于题目的例子:对于 A = [1,2,1,2,3], K = 2,我们运行上面的代码过程中,会得到的两个满足题意的子数组 [1, 2, 1, 2]和[2, 3]。
对于子数组[1, 2, 1, 2],它的所有子数组都是满足题意的,共有 1 + 2 + 3 + 4 = 10 个子数组。
以第一个 1 为右端点的满足题意的子数组为 [1];
以第一个 2 为右端点的满足题意的子数组为 [1,2], [2];
以第二个 1 为右端点的满足题意的子数组为 [1,2,1],[2,1], [1];
以第二个 2 为右端点的满足题意的子数组为 [1,2,1,2], [2,1,2],[1,2], [2];
对于子数组[2, 3],它的所有子数组都满足题意,共有 3 个子数组。
以 2 为右端点的满足题意的子数组,在上面已经统计过了,因此不要重复统计。
以 3 为右端点的满足题意的子数组为[2, 3], [3]。
所以总的数组 A 有 12 个由 最多 2 个不同整数组成的子数组。
所以,当 right 到达一个新位置之后,把 left 调整到满足题意的位置,当前[left, right]区间内符合条件的并且以 right 为右端点的子数组个数 为 right - left + 1。当 right 指针把数组的每个位置遍历一遍,就得到了以每个位置作为区间右短点的子数组长度,累加得到的就是结果。
作者:fuxuemingzhu
链接:https://leetcode-cn.com/problems/subarrays-with-k-different-integers/solution/cong-zui-jian-dan-de-wen-ti-yi-bu-bu-tuo-7f4v/
以我个人愚见,通过这种方法好求个数的原因是不同整数个数小于k的要求很松导致它的左右指针可以移动,方便跟着这个窗口去求个数。如果定死了个数,也就定死了这个窗口,中间很多很细的子数组也直接略过去了。
下面给出代码
int GetMostDistinct(int *nums,int numsSize,int k) {//此函数是求最多不超过k个不同数的子串的个数
int num1[20001],cnt=0;//直接记录不同数的个数
memset(num1, 0, sizeof(int) * (numsSize + 1));
int left = 0, right = 0, ret = 0;
while (right < numsSize) {
if (num1[nums[right]] == 0)
cnt++;
++num1[nums[right++]];
while (cnt > k) {
--num1[nums[left]];
if (num1[nums[left]] == 0)
cnt--;
++left;
}
ret += right - left;//这里是因为上面写了right++。如果right++写在while循环的最底下,那这里要改成right-left+1
}
return ret;
}
int subarraysWithKDistinct(int *nums,int numsSize, int K) {
return GetMostDistinct(nums,numsSize, K) - GetMostDistinct(nums,numsSize, K - 1);}
这道题给了我很多启发:
首先就是求子数组的方法,定右端点,然后根据窗口的大小来进行累加
然后就是转换思想,通过前缀和的方式来转化所求的目标