Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Python의 목록에 있는 요소 x 및 x+1 계산

<시간/>

nums라는 숫자 목록이 있다고 가정하고 x + 1도 존재하도록 x가 있는 요소의 수를 찾아야 합니다.

따라서 입력이 [2, 3, 3, 4, 8]과 같으면 출력은 3

이 됩니다.

이 문제를 해결하기 위해 다음 단계를 따릅니다. −

  • s :=숫자에 있는 요소를 삽입하여 집합을 만듭니다.
  • 카운트:=0
  • 숫자 단위의 각 i에 대해 다음을 수행합니다.
    • s에서 i+1이면
      • 카운트 :=카운트 + 1
  • 반환 횟수

이해를 돕기 위해 다음 구현을 살펴보겠습니다. −

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