计算下一个元素也在 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