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

（例如，[1,2,3,1,2] 中有 3 個不同的整數：1，2，以及 3。）

返回 A 中好子數組的數目。

示例 1：

輸入：A = [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].

解題思路

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

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。