Đếm phần tử x và x+1 cùng tồn tại trong danh sách Python

Cho danh sách nums, đếm số phần tử x sao cho x+1 cũng có trong danh sách.

Ví dụ: [2, 3, 3, 4, 8] → kết quả 3 (2 có 3, 3 có 4, 3 có 4).

Các bước giải quyết

• s = set(nums)
• count = 0
• Với mỗi i trong nums: nếu i+1 in s thì count += 1
• Trả về count

Ví dụ mã Python

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

Đầu vào

[2, 3, 3, 4, 8]

Đầu ra

3