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
- 如果 i+1 在 s 中,则
- 返回 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
广告