leetcode怎么解決K 個不同整數的子數組

這篇文章主要介紹“leetcode怎么解決K 個不同整數的子數組”，在日常操作中，相信很多人在leetcode怎么解決K 個不同整數的子數組問題上存在疑惑，小編查閱了各式資料，整理出簡單好用的操作方法，希望對大家解答”leetcode怎么解決K 個不同整數的子數組”的疑惑有所幫助！接下來，請跟著小編一起來學習吧！

一、題目內容

給定一個正整數數組 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].

示例 2：

輸入：A = [1,2,1,3,4], K = 3
輸出：3
解釋：恰好由 3 個不同整數組成的子數組：[1,2,1,3], [2,1,3], [1,3,4].

提示：

1 <= A.length <= 20000
1 <= A[i] <= A.length
1 <= K <= A.length

二、解題思路

最多存在 K 個不同整數的子區間的個數與恰好存在K個不同整數的子區間的個數的差恰好等于最多存在 K - 1個不同整數的子區間的個數。

因此，最多存在 K 個不同整數的子區間的個數 - 最多存在 K - 1個不同整數的子區間的個數 = 恰好存在K個不同整數的子區間的個數。

三、代碼

class Solution:
    def subarraysWithKDistinct(self, A: list, K: int) -> int:
        return self._subarrayWithDistinct(A, K) - self._subarrayWithDistinct(A, K - 1)

    # 計算最多為K的子數組長度
    def _subarrayWithDistinct(self, A, K):
        freq_num = [0 for _ in range(len(A) + 1)]

        left, right = 0, 0
        count = 0
        res = 0

        while right < len(A):
            if freq_num[A[right]] == 0:
                count += 1
            freq_num[A[right]] += 1
            right += 1

            while count > K:
                freq_num[A[left]] -= 1
                if freq_num[A[left]] == 0:
                    count -= 1
                left += 1
            # 長度貢獻
            res += right - left
        return res


if __name__ == '__main__':
    s = Solution()
    A = [1, 2, 1, 2, 3]
    K = 2
    ans = s.subarraysWithKDistinct(A, K)
    print(ans)

到此，關于“leetcode怎么解決K 個不同整數的子數組”的學習就結束了，希望能夠解決大家的疑惑。理論與實踐的搭配能更好的幫助大家學習，快去試試吧！若想繼續學習更多相關知識，請繼續關注億速云網站，小編會繼續努力為大家帶來更多實用的文章！