Pythonで正確にk個の一意の要素を持つサブリストの数をカウントするプログラム

numsと呼ばれる番号のリストと別の値kがあるとすると、サブリストに正確にk個の一意の番号が存在するために必要なサブリストの数を見つける必要があります。

したがって、入力がnums =[2、2、3、4] k =2のような場合、[2、2、3]、[2、3]、のようなサブリストがあるため、出力は3になります。 [3、4]。

これを解決するには、次の手順に従います-

• 関数count（）を定義します。これにはKがかかります
• slot：=空のマップ、デフォルトではすべての値は0です
• i：=res：=0
• 各インデックスj、および値x numに対して、do
  • slot [x]：=slot [x] + 1
  • スロットのサイズ>Kの場合、
    • slot [nums [i]]：=slot [nums [i]]-1
    • slot [nums [i]]が0と同じ場合、
      • スロットを削除[nums[i]]
    • i：=i + 1
  • res：=res + j-i + 1
• return res
• メインの方法から次のようにします-
• return count（k）-count（k-1）

例（Python）

理解を深めるために、次の実装を見てみましょう-

from collections import Counter
class Solution:
   def solve(self, nums, k):
      def count(K):
         slot = Counter()
         i = res = 0
         for j, x in enumerate(nums):
            slot[x] += 1
            while len(slot) > K:
               slot[nums[i]] -= 1
               if slot[nums[i]] == 0:
                  del slot[nums[i]]
               i += 1
            res += j - i + 1
         return res
      return count(k) - count(k - 1)
ob = Solution()
nums = [2, 2, 3, 4]
k = 2
print(ob.solve(nums, k))

入力

[2, 2, 3, 4], 2

出力

3