[hackerrank] Sherlock and Array python

 

 

무난하게 풀었던 것같다.

 

1.  먼저 arr의 totalSum을 구하고, leftSum, rightSum을 초기화 해주자

 

2. for문을 돌려서 i > 0일때 leftSum += arr[i-1], rightSum = totalSum - arr[i] - leftSum 즉, 인덱스가 하나씩 늘어날때마다 leftSum과 rightSum을 구해주자.

 

3. leftSum == rightSum 이면 return YES for문이 다돌면 return NO

 

 

def balancedSums(arr):
    # Write your code here
    totalSum = sum(arr)
    leftSum = 0
    rightSum = 0
    for i in range(len(arr)):
        if i > 0: 
            leftSum += arr[i-1]
        rightSum = totalSum - arr[i] - leftSum
        if leftSum == rightSum:
            return 'YES'
        else:
            rightSum = 0
    return 'NO'