집합에 요소의 참여를 확인하여 행복을 찾는 Python 프로그램

<시간/>

n개의 다른 정수를 가진 배열 num이 있다고 가정합니다. 또한 두 개의 분리된 집합 A와 B가 있습니다. 처음에는 0으로 설정된 행복 매개변수가 하나 있습니다. 우리는 num의 각 정수 i를 살펴봅니다. i가 A에 있으면 행복을 1 더하고 i가 B에 있으면 1을 줄입니다. 마침내 최종 행복 값을 찾아야 합니다.

따라서 입력이 nums =[1,2,5,8,6,3] A ={5,8,9,7,3} B ={2,4,12,15}인 경우 출력은 5, 8, 3이 A에 있으므로 현재 행복은 3이지만 2는 B에 있으므로 1로 줄이면 행복은 2가 되므로 2가 됩니다.

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

행복:=0
숫자 단위의 각 i에 대해 다음을 수행합니다.
- 내가 A에 있으면
  - 행복 :=행복 + 1
- 그렇지 않으면 i가 B에 있을 때
  - 행복 :=행복 - 1
행복함

예시

더 나은 이해를 위해 다음 구현을 살펴보겠습니다.

def solve(nums, A, B):
   happiness = 0
   for i in nums:
      if i in A:
         happiness += 1
      elif i in B:
         happiness -= 1
   return happiness

nums = [1,2,5,8,6,3]
A = {5,8,9,7,3}
B = {2,4,12,15}
print(solve(nums, A, B))

입력

[1,2,5,8,6,3], {5,8,9,7,3}, {2,4,12,15}