পাইথনে অ্যারেতে যেকোনো এলিমেন্ট বাকি সব এলিমেন্টের যোগফলের সমান কিনা

একটি অ্যারে nums দেওয়া আছে। যাচাই করতে হবে যে কি কোনো একটি এলিমেন্ট আছে যার মান বাকি সব এলিমেন্টের যোগফলের সমান।

উদাহরণ

ইনপুট: nums = [3, 2, 10, 4, 1]

আউটপুট: True

ব্যাখ্যা: ১০ = ৩+২+৪+১।

অ্যালগরিদম

1. মোট যোগফল total বের করুন।
2. যদি total জোড় হয়, তবে total/2 মানটি অ্যারেতে আছে কিনা চেক করুন।

উদাহরণ কোড

from collections import defaultdict

def solve(nums):
    freq = defaultdict(int)
    total = 0
    for x in nums:
        freq[x] += 1
        total += x
    if total % 2 == 0:
        if freq[total // 2]:
            return True
    return False

nums = [3, 2, 10, 4, 1]
print(solve(nums))  # Output: True

ইনপুট

[3, 2, 10, 4, 1]

আউটপুট

True