Suppose we have an array called nums we have to check whether the array contains an element whose value is same as sum of all other elements.
So, if the input is like nums = [3,2,10,4,1], then the output will be True, 10 = (3+2+4+1).
To solve this, we will follow these steps −
Let us see the following implementation to get better understanding −
from collections import defaultdict def solve(nums): freq = defaultdict(int) total = 0 for i in range(len(nums)): freq[nums[i]] += 1 total += nums[i] if total % 2 == 0: if freq[total // 2]: return True return False nums = [3,2,10,4,1] print(solve(nums))