পাইথনে সেটে এলিমেন্টের অংশগ্রহণ চেক করে হ্যাপিনেস বের করার প্রোগ্রাম

ধরুন আমাদের একটি অ্যারে nums আছে যাতে nটি ভিন্ন ইন্টিজার রয়েছে। আমাদের আরেকটি দুটি ডিজয়েন্ট সেট A এবং B আছে। আমাদের একটি হ্যাপিনেস প্যারামিটার আছে যা প্রাথমিকভাবে ০। আমরা nums-এর প্রতিটি ইন্টিজার i এর জন্য যাব। যদি i A-তে থাকে, তবে হ্যাপিনেস ১ বেড়াব এবং যদি B-তে থাকে, তবে ১ কমিয়ে দিব। চূড়ান্ত হ্যাপিনেস ভ্যালু বের করতে হবে।

উদাহরণস্বরূপ, যদি ইনপুট nums = [1,2,5,8,6,3], A = {5,8,9,7,3}, B = {2,4,12,15} হয়, তবে আউটপুট হবে 2 কারণ ৫, ৮, ৩ হলো A-তে তাই হ্যাপিনেস ৩, কিন্তু ২ হলো B-তে তাই ১ কমিয়ে ২ হয়ে গেল।

সমাধানের ধাপসমূহ

happiness := 0
প্রতিটি i ইন nums এর জন্য:
- যদি i in A, তবে happiness += 1
- অন্যথায় যদি i in B, তবে happiness -= 1
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}