程序查找在Python中第一个和最后一个值相同的子列表

假设我们有一个称为nums的数字列表，我们必须找到第一个元素和最后一个元素相同的子列表的数量。

因此，如果输入类似于nums = [10、15、13、10]，则输出将为5，因为具有相同的第一个和最后一个元素的子列表是：[10]，[15]，[13]，[ 10]，[10、15、13、10]。

为了解决这个问题，我们将遵循以下步骤-

• num_sublists：= nums的大小

• d：=一个空的映射

• 对于每n个数字，执行

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

• 对于d中元素的每个数字k和对应的频率v，

• num_sublists：= num_sublists +（(v-1)*(v)/ 2的商）

• 如果v与1不相同，则

• 返回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