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