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。
示例
让我们看下面的实现来更好地理解
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