[python] Codility Lesson 5-4. PassingCars

 

앞선 문제들과 비교해 비교적 빨리, 무난히 마음에 들게 잘 푼거 같다.

 

def solution(A):
    # write your code in Python 3.6
    sum = 0
    east_car = 0
    for i in range(len(A)):
        if A[i] == 0:
            east_car += 1
        else:
            sum += 1 * east_car 
            if sum > 1000000000:
                sum = -1
                break

    return sum

 

1. 동쪽으로 가는 방향의 차량을 for문을 돌면서 만날때 마다 1씩 더해주자.

 

2. 서쪽으로 가는 방향의 차량을 for문을 돌면서 만날때 마다 sum에다 1 * east_car를 해주자.

 

3. 문제에서 주어진 예외를 처리해주자.