首页 文章资讯内容详情

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

2026-06-03 1 花语

假设我们有一个数字列表,比如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