[Algorithm] PassingCars

[hwangJi-dev]

💬 문제

https://app.codility.com/programmers/lessons/5-prefix_sums/passing_cars/

💬 Idea

• P가 Q보다 작을 때만 순서쌍이 만들어진다는 원리를 이용하여 풀어주었다.

1. 먼저 서쪽으로 향하는 차의 개수를 구하여 westCarCount에 저장해놓는다.
2. 이후 반복문을 돌면서
   1. 차가 서쪽으로 향할 때마다 westCarCount를 -1씩 차감해주고
   2. 동쪽으로 향하는 차가 나오면 당시의 westCarCount를 ans에 더하여 순서쌍의 개수를 도출해주었다.

💬 풀이

public func solution(_ A : inout [Int]) -> Int {
    var westCarCount = A.filter({ $0 == 1 }).count
    var ans = 0
    
    for a in A {
        if a == 0 {
            ans += westCarCount
        } else {
            westCarCount -= 1
        }
    }
    
    return ans > 1000000000 ? -1 : ans
}

소요시간 : 12분

시간 복잡도 : O(N)

평가표 : https://app.codility.com/demo/results/trainingYDRNDZ-VG7/