給定一個正整數數組 A,如果 A 的某個子數組中不同整數的個數恰好為 K,則稱 A 的這個連續、不一定獨立的子數組為好子數組。
(例如,[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].
解題思路
這題與普通滑動窗口的不同之處在于:要統計[l,r]區間內的滿足不同整數的個數恰好為 K的子數組的個數,按照常規滑動窗口做法需要不斷l++,直到不同整數的個數不為 K,但是這樣的話,新得到的窗口就會丟失一部分元素,后面枚舉r的時候,就會有遺漏。
因此這題不直接移動l,而是選擇維護一個多重的滑動窗口,[l…l1…r]其中[l1,r]區間為滿足不同整數的個數恰好為 K-1的區間,
設區間末尾都是r
因為[l1…r]是恰好只有k-1個不同整數的,所以起始位置在l1前面的子數組,區間內的不同整數的個數都大于k-1
因為[l…r]是恰好只有k個不同整數的,所以起始位置在l前面的子數組,區間內的不同整數的個數都大于k
綜上所述,起始位置處于l和l1的子數組,區間內的不同整數的個數都剛好等于k
因此統計[l,r]區間內的滿足不同整數的個數恰好為 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;}
}