用 Python 编写一个程序来计算数组中下一个元素也存在的元素

pythonserver side programmingprogramming更新于 2025/12/17 17:48:17

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

因此，如果输入为 nums = [4, 2, 3, 3, 7, 9]，则输出将为 3，因为 2+1 = 3 存在，3+1 = 4 存在，另外 3 存在，所以总数为 3。

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

answer := 0
c := 包含 nums 中每个元素出现频率的列表
dlist := 包含所有键的列表c
对于 dlist 中的每个 i，执行
- 如果 c[i + 1] > 0，则
  - answer := answer + c[i]
返回 answer

示例

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

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))