通过检查元素是否参与集合来寻找幸福的 Python 程序
pythonserver side programmingprogramming更新于 2025/12/21 17:16:17
假设我们有一个包含 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 中时，则
- happiness := happiness - 1
- 如果 i 在 A 中，则
- 返回 happiness
示例
让我们看看下面的实现以便更好地理解
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
相关文章
有用资源
python 参考教程 - 该教程包含有关 python 的更多信息:https://www.cainiaomax.com/python/
