计算下一个元素也在 Python 数组中的元素的程序

假设我们有一个数字列表,比如 nums,我们必须找到数组中元素的数量 x,这样 x + 1 也存在于数组中。

因此,如果输入类似于 nums = [4, 2, 3, 3, 7, 9],那么输出将是 3,因为存在 2+1 = 3,存在 3+1 = 4 并且另外 3 是现在总共 3。

示例

让我们看下面的实现来更好地理解

from collections import Counter

def solve(nums):

   answer = 0

   c = Counter(nums)

   dlist = list(c.keys())

   for i in dlist:

      if c[i + 1] > 0:

         answer += c[i]

   return answer

nums = [4, 2, 3, 3, 7, 9]

print(solve(nums))

输入

[4, 2, 3, 3, 7, 9]
输出结果
3

以上是 计算下一个元素也在 Python 数组中的元素的程序 的全部内容, 来源链接: utcz.com/z/354375.html

回到顶部