Lesson 4: Counter Elements → Max Counters

문제

https://app.codility.com/programmers/lessons/4-counting_elements/max_counters/

리스트의 값이 N이하 일 경우에는, 해당하는 인덱스의 값을 1 증가시키고 N보다 큰 값인 경우에는 모든 인덱스 값을 현재 값의 max로 변경한다.

문제 풀이

각 인덱스에 값을 추가하다가, N보다 큰 값이 발생한 경우 answer = max(answer) * N로 처리하였더니 효율성을 통과하지 못하였다.

def solution(N, A):
    answer = [0] * N
    for idx in A:
        print(answer)
        if idx > N:
            answer = [max(answer)] * N
        else:
            answer[idx - 1] += 1

    return answer

코드

def solution(N, A):
    answer = [0] * N
    next_max_cnt = max_cnt = 0

    for idx in A:
        if idx <= N:
            current = answer[idx - 1] = max(answer[idx - 1] + 1, max_cnt + 1)
            next_max_cnt = max(current, next_max_cnt)
        else:
            max_cnt = next_max_cnt

    return [cnt if cnt > max_cnt else max_cnt for cnt in answer]

• 위의 코드와 달리 max 값을 찾으면 즉각적으로 반영하지 않는다.
• 모든 과정이 끝난 후에 cnt 값과 max_cnt 값을 비교하여 처리하면 효율성을 높일 수 있다.