Leikou問題ブラッシングノート:992。異なる整数のKサブ配列(スライディングウィンドウメソッド、わかりやすいコード、完全なソリューションコードとコメント)(暴力的なソリューション、Pythonタイムアウトの結果...)

トピック:

992、異なる整数のK個のサブ配列

正の整数Aの配列が与えられ、Aのサブ配列内の異なる整数の数がたまたまKである場合、Aのこの連続的で、必ずしも独立していないサブ配列は、適切なサブ配列と呼ばれます。

(たとえば、[1,2,3,1,2]には1、2、および3の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 + 1個の異なる整数を持つサブ配列の数

サブ問題を解決する:

(1)iを列挙し、iで始まり、少なくともK個の異なる整数で始まるサブ配列の数を見つけます。
(2)[i、j]が条件を満たすサブ配列であるような修飾されたjを見つけた場合、[ i、j]、[i、j + 1]、[i、j + 2]、...、[i、n-1]は、すべて条件を満たすサブ配列です。
(3)その後、ウィンドウ内の配列が条件を満たさなくなるまでウィンドウを縮小し、ウィンドウを再度拡大して前の手順に戻ります。

問題解決のPythonコード:

class Solution:
    def subarraysWithKDistinct(self, A: List[int], K: int) -> int:
        def helper(A,K):
            n = len(A)
            i = 0
            res = 0
            ########################## 
            # 窗口内需要维护的变量
            diff_nums = 0
            counter = collections.defaultdict(int)
            ##########################
            for j in range(n):
                # 扩大窗口
                counter[A[j]] += 1
                if counter[A[j]] == 1:
                    diff_nums += 1

                # 如果找到以i开头满足条件的子数组了,就更新答案并缩小窗口
                while diff_nums > K:
                    res += n-j
                    counter[A[i]] -= 1
                    if counter[A[i]] == 0:
                        diff_nums -= 1
                    i += 1
            return res
        
        return -helper(A,K) + helper(A,K-1)

ここに画像の説明を挿入

著者:MiloMusiala
リンク:https://leetcode-cn.com/problems/subarrays-with-k-different-integers/solution/python-dong-hua-tong-xiang-shuang-zhi-zh-57ym/
出典:force控除(LeetCode)https://leetcode-cn.com/problems/subarrays-with-k-different-integers/

暴力的な解決策のアイデア:

Aのすべてのサブ配列を取得し、各サブ配列を取得するときにその異なる整数がKに等しいかどうかを判別し、最後に条件を満たすサブ配列の数を返します。

シンプルで失礼ですが、残念ながら残業です。

参考のために、実際のプロジェクトで使用できます

暴力的なソリューションのPythonコード:

class Solution:
    def subarraysWithKDistinct(self, A: List[int], K: int) -> int:
        ans = 0
        n = len(A)
        # x + 1 表示子串的长度
        for x in range(n):
            # i 表示滑窗长度
            for i in range(x+1,n+1):
                # if len(set(A[x:i]))==K:
                #     ans += 1
                if len(set(A[x:i]))==K:
                    ans += 1
        return ans

おすすめ

転載: blog.csdn.net/weixin_44414948/article/details/113771127