Cho mảng nums và hai tập hợp rời rạc A, B. Khởi tạo happiness = 0. Duyệt nums: nếu phần tử trong A thì +1, nếu trong B thì -1. Trả về giá trị cuối cùng.
Ví dụ: nums = [1,2,5,8,6,3], A = {5,8,9,7,3}, B = {2,4,12,15} → 2.
Ví dụ mã nguồn
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)) # 2
Đầu vào / Đầu ra
Input: [1,2,5,8,6,3], {5,8,9,7,3}, {2,4,12,15}
Output: 2