loadbalancer.py

def solution(A):
    if len(A) < 5:
        return False

    total_s = sum(A)
    left_s,right_s = A[0],A[-1]
    left_idx,right_idx = 1,len(A) - 2

    while left_idx < right_idx:
        mid_s = total_s - left_s - right_s - (A[left_idx] + A[right_idx])
        if mid_s < left_s or mid_s < right_s:
            return False
        if left_s == mid_s == right_s:
            return True

        if left_s < right_s:
            left_s += A[left_idx]
            left_idx += 1
        else:
            right_s += A[right_idx]
            right_idx -= 1

    return True


if __name__ == '__main__':
    results = solution([1, 3, 4, 2, 2, 2, 1, 1, 2])
    print(results)