# Check if the array has an element which is equal to sum of all the remaining elements in Python

PythonServer Side ProgrammingProgramming

#### Beyond Basic Programming - Intermediate Python

Most Popular

36 Lectures 3 hours

#### Practical Machine Learning using Python

Best Seller

91 Lectures 23.5 hours

#### Practical Data Science using Python

22 Lectures 6 hours

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 −

• freq := an empty map
• total := 0
• for i in range 0 to size of nums - 1, do
• freq[nums[i]] := freq[nums[i]] + 1
• total := total + nums[i]
• if total is even, then
• if freq[quotient of (total / 2)] is non-zero, then
• return True
• return False

Let us see the following implementation to get better understanding −

## Example Code

Live Demo

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))

## Input

[3,2,10,4,1]

## Output

True