K个不同整数的子数组(纯C)

写力扣遇见的题,思想方法很好,记录一下。

题目:

给定一个正整数数组 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 *10^4
  • 1 <= nums[i], k <= nums.length

这个题目有几个问题需要解决:

怎么知道当前子数组有多少不同整数?

怎么把那些很细的子数组也给算到我的总和里?

我们用一个变量cnt来记录当前子数组有多少不同整数,用双指针来进行遍历。如果nums[right]没有出现过,cnt++,如果nums[left]出现的此数为1且left将要移动,cnt--。

那怎么知道这个数有没有出现过?只能用数组来记录了(只是单纯因为我不会哈希表)。

这时候我们想,如果cnt刚好等于k,左右指针都停下来的时候,我们要怎么计算这一段区间的子数组呢?

如果你擅长找规律,或者排列组合学的很好,你应该发现了,子数组的个数等于\frac{(length+1)*length}{2}(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);}

这道题给了我很多启发:

首先就是求子数组的方法,定右端点,然后根据窗口的大小来进行累加

然后就是转换思想,通过前缀和的方式来转化所求的目标

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值