[프로그래머스] 캐시 - 파이썬 풀이

나의 코드

def solution(cacheSize, cities):
    answer = 0
    data = []

    if cacheSize == 0:
        return len(cities) * 5

    for i in cities: 
        i = i.lower()
        if i not in data:
            if len(data) == cacheSize:
                data.pop(0)
            data.append(i)
            answer += 5

        else:
            data.pop(data.index(i))
            data.append(i)
            answer += 1

    return answer

 

다른 코드

def solution(cacheSize, cities):
    import collections
    cache = collections.deque(maxlen=cacheSize)
    time = 0
    for i in cities:
        s = i.lower()
        if s in cache:
            cache.remove(s)
            cache.append(s)
            time += 1
        else:
            cache.append(s)
            time += 5
    return time

deque의 maxlen 인자를 이용