Codility - Max Counter

max-counter.py

def solution(N, A):
    max_counter = last_max_counter = 0
    counters = [0] * N

    for k in xrange(len(A)):
        current_counter = A[k]

        if current_counter == N+1:
            last_max_counter = max_counter
        else:
            counters[current_counter - 1] = max(counters[current_counter - 1], last_max_counter) + 1
            max_counter = max(max_counter, counters[current_counter - 1])

    for i in xrange(N):
        counters[i] = max(counters[i], last_max_counter)

    return counters