通过检查元素是否在集合中来寻找幸福感的Python程序

假设我们有一个具有n个不同整数的数组nums。我们还有两个不相交的集合A和B。我们有一个幸福参数，最初设置为0。我们遍历nums中的每个整数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}，则输出将为2，因为5、8和3在A中，所以幸福感现在为3，但2在B中，所以将其减1后，幸福感为2。

为了解决这个问题，我们将遵循以下步骤

• 幸福感:=0
• 对于nums中的每个i，执行以下操作
  • 如果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}

输出

2