Pythonで最初と最後の値が同じであるサブリストを見つけるためのプログラム

numsという番号のリストがあるとすると、最初の要素と最後の要素が同じであるサブリストの数を見つける必要があります。

したがって、入力がnums =[10、15、13、10]の場合、同じ最初と最後の要素を持つサブリストは[10]、[15]、[13]、[ 10]、[10、15、13、10]。

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

• num_sublists：=numsのサイズ

• d：=空のマップ

• numsのnごとに、実行します

  • d [n]：=d [n] + 1

• dの要素の数kと対応する周波数vごとに、実行します

  • vが1と同じでない場合、

    • num_sublists：=num_sublists +（（v-1）*（v）/ 2の商）

• num_sublistsを返す

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

例

from collections import defaultdict
class Solution:
   def solve(self, nums):
      num_sublists = len(nums)

      d = defaultdict(int)
      for n in nums:
         d[n] += 1
   
      for k,v in d.items():
         if v != 1:
            num_sublists += (v-1)*(v)//2
      return num_sublists
ob = Solution()
nums = [10, 15, 13, 10]
print(ob.solve(nums))

入力

[10, 15, 13, 10]

出力

5