alternating Sums

alternating Sums

Several people are standing in a row and need to be divided into two teams. The first person goes into team 1, the second goes into team 2, the third goes into team 1 again, the fourth into team 2, and so on.

You are given an array of positive integers – the weights of the people. Return an array of two integers, where the first element is the total weight of team 1, and the second element is the total weight of team 2 after the division is complete.

Example

For a = [50, 60, 60, 45, 70], the output should be
solution(a) = [180, 105].

Input/Output

  • [execution time limit] 4 seconds (py3)
  • [input] array.integer aGuaranteed constraints:
    1 ≤ a.length ≤ 105,
    45 ≤ a[i] ≤ 100.
  • [output] array.integer

 

def solution(a):
    total_team1 = 0
    total_team2 = 0
    teams = []
   
    for i in range(0, len(a)):
        if i % 2 == 0:
            total_team1 += a[i]
        else:
            total_team2 += a[i]
   
    teams.append(total_team1)
    teams.append(total_team2)
    return teams
No Comments

Post A Comment