codility Lesson4 - MaxCounters

코드

시간복잡도: O(N + M)

// 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;

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

    vector<int> arr(N, 0);
    int max = 0, memo = 0;
    for(auto a: A) {
        if(N > a-1) {
            arr[a-1] = memo > arr[a-1] ? memo + 1 : arr[a-1] + 1;
            max = arr[a-1] > max ? arr[a-1] : max;
        } else {
            memo = max;
        }
    }

    for(int i=0;i<arr.size();i++) {
        if(memo > arr[i]) arr[i] = memo;
    }

    return arr;
}

풀이 및 소감

처음에는 for문 안 에서 fill을 사용하여 최대 값으로 갱신해줬었는데 역시나 시간 초과가 났다. 그래서 max와 비교해서 arr 값을 변경시켜줄까 하다가 마지막에 max로 올릴 때 값이 맞지 않을 거 같아서 memo라는 변수를 사용해서 해결했다. 다른 사람들의 풀이도 검색해봤는데 비슷하게 푼 거 같다!