codility Lesson5 - PassingCars

코드

O(N)

// you can use includes, for example:
// #include <algorithm>

// you can write to stdout for debugging purposes, e.g.
// cout << "this is a debug message" << endl;

int solution(vector<int> &A) {
    // write your code in C++14 (g++ 6.2.0)

    int count = 0, answer = 0;
    for(int i=A.size()-1; i>=0;i--) {
        if(A[i] == 1) count++;
        else answer += count;

        if(answer > 1000000000) return -1;
    }

    return answer;
}

풀이 및 소감

문제 이해가 가장 어려웠다... passing을 그냥 지나갔다라고 생각해서 어디서 지나갔는지 먼가 이해가 안됐었다. 그래서 결국 다른 사람들 풀이는 안 보고 문제 해석한 것을 보고 해결했다. 문제 이해한다고 너무 신경써서 for문을 역순으로 했는데 굳이 그렇게 안 해도 됐을 거 같다. 1의 개수를 세리면서 0을 만나면 답에 더해주는 식으로 해결했다. 문제 이해가 너무 어려웠던 문제 ㅠㅠ