Python 列表中统计元素 x 和 x+1 的个数


假设我们有一个名为 nums 的数字列表,我们需要找到其中存在 x 和 x + 1 的元素个数。

例如,如果输入是 [2, 3, 3, 4, 8],则输出为 3

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

  • s := 创建一个集合,并将 nums 中的元素插入其中
  • count := 0
  • 对于 nums 中的每个 i:
    • 如果 i+1 在 s 中,则
      • count := count + 1
  • 返回 count

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

示例

实时演示

class Solution:
   def solve(self, nums):
      s = set(nums)
      count = 0
      for i in nums:
         if i+1 in s:
            count += 1
      return count
ob = Solution()
nums = [2, 3, 3, 4, 8]
print(ob.solve(nums))

输入

[2, 3, 3, 4, 8]

输出

3

更新于:2020年9月22日

123 次浏览

开启你的职业生涯

完成课程获得认证

开始学习
广告