Computer >> Máy Tính >  >> Lập trình >> Python

Đếm phần tử x và x + 1 có trong danh sách bằng Python

Giả sử chúng ta có một danh sách các số được gọi là num, chúng ta phải tìm số phần tử x có sao cho x + 1 cũng tồn tại.

Vì vậy, nếu đầu vào là [2, 3, 3, 4, 8], thì đầu ra sẽ là 3

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau -

  • s:=tạo một tập hợp bằng cách chèn các phần tử có trong nums
  • số lượng:=0
  • đối với mỗi tôi trong nums, hãy thực hiện
    • nếu tôi +1 trong giây, thì
      • count:=count + 1
  • số lần trả lại

Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn -

Ví dụ

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