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

Arnab Chakraborty

更新於：2020年9月22日

123 次瀏覽

開啟你的職業生涯

完成課程獲得認證

開始學習