給定一個(gè)正整數(shù)數(shù)組 A，如果 A 的某個(gè)子數(shù)組中不同整數(shù)的個(gè)數(shù)恰好為 K，則稱(chēng) A 的這個(gè)連續(xù)、不一定獨(dú)立的子數(shù)組為好子數(shù)組。

（例如，[1,2,3,1,2] 中有 3 個(gè)不同的整數(shù)：1，2，以及 3。）

返回 A 中好子數(shù)組的數(shù)目。

示例 1：

輸入：A = [1,2,1,2,3], K = 2
輸出：7
解釋：恰好由 2 個(gè)不同整數(shù)組成的子數(shù)組：[1,2], [2,1], [1,2], [2,3], [1,2,1], [2,1,2], [1,2,1,2].

解題思路

這題與普通滑動(dòng)窗口的不同之處在于：要統(tǒng)計(jì)[l,r]區(qū)間內(nèi)的滿(mǎn)足不同整數(shù)的個(gè)數(shù)恰好為 K的子數(shù)組的個(gè)數(shù)，按照常規(guī)滑動(dòng)窗口做法需要不斷l(xiāng)++,直到不同整數(shù)的個(gè)數(shù)不為 K，但是這樣的話(huà)，新得到的窗口就會(huì)丟失一部分元素，后面枚舉r的時(shí)候，就會(huì)有遺漏。
因此這題不直接移動(dòng)l，而是選擇維護(hù)一個(gè)多重的滑動(dòng)窗口，[l…l1…r]其中[l1,r]區(qū)間為滿(mǎn)足不同整數(shù)的個(gè)數(shù)恰好為 K-1的區(qū)間，
設(shè)區(qū)間末尾都是r
因?yàn)閇l1…r]是恰好只有k-1個(gè)不同整數(shù)的，所以起始位置在l1前面的子數(shù)組，區(qū)間內(nèi)的不同整數(shù)的個(gè)數(shù)都大于k-1
因?yàn)閇l…r]是恰好只有k個(gè)不同整數(shù)的，所以起始位置在l前面的子數(shù)組，區(qū)間內(nèi)的不同整數(shù)的個(gè)數(shù)都大于k
綜上所述，起始位置處于l和l1的子數(shù)組，區(qū)間內(nèi)的不同整數(shù)的個(gè)數(shù)都剛好等于k
因此統(tǒng)計(jì)[l,r]區(qū)間內(nèi)的滿(mǎn)足不同整數(shù)的個(gè)數(shù)恰好為 K的子數(shù)組的個(gè)數(shù)，只需要計(jì)算l1-l即可。

代碼

class Solution {public int subarraysWithKDistinct(int[] A, int K) {int n=A.length;int[] cnt1=new int[n+1],cnt2=new int[n+1];int n1=0,n2=0,l1=0,l2=0;int l=0,r=0,res=0;while (r<n){if(cnt1[A[r]]==0)n1++;cnt1[A[r]]++;if(cnt2[A[r]]==0)n2++;cnt2[A[r]]++;while (n2>K-1){cnt2[A[l2]]--;if(cnt2[A[l2]]==0)n2--;l2++;}while (n1>K){cnt1[A[l1]]--;if(cnt1[A[l1]]==0)n1--;l1++;}res+=l2-l1;r++;}return res;} }

總結(jié)

以上是生活随笔為你收集整理的leetcode 992. K 个不同整数的子数组（滑动窗口）的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。