[codility] Lesson 3. Time Complexity - TapeEquilibrium

https://app.codility.com/programmers/lessons/3-time_complexity/tape_equilibrium/

TapeEquilibrium coding task - Learn to Code - Codility

 

• N개의 정수로 이루어진 배열 A
• A[0], A[1], ..., A[P − 1]으로 합쳐진 값과 A[P], A[P + 1], ..., A[N − 1]으로 합쳐진 값들을 나눈 절대값을 구한다.
• 절대값들 중에서 가장 최소값을 출력한다.
• O(N)

def solution(A):
    res = float('inf')
    sum1 = 0
    sum2 = sum(A)
    for i in range(1,len(A)):
        sum1 += A[i-1]
        sum2 -= A[i-1]
        res = min(res, abs(sum1-sum2))
    return res