[코딩테스트] 프로그래머스 프린터 파이썬(Python)

https://school.programmers.co.kr/learn/courses/30/lessons/42587

프로그래머스

from collections import deque
def solution(priorities, location):
    temp = [(p, i) for i, p in enumerate(priorities)]
    temp = my_sort(temp)
    for i in range(len(temp)):
        if location == temp[i][1]:
            return i+1

def my_sort(temp):
    q =deque(temp)
    result = []
    while q:
        j, index = q.popleft()
        for i in range(0, len(q)):
            if q[i][0] > j:
                q.append((j, index))
                break
        else:
            result.append((j, index))
    return result

먼저 저는 list comprehension을 사용하여 temp에 (우선순위, 인덱스)형태로 값을 저장하였습니다.

 

my_sort함수를 작성하여 문제에서 요구한 인쇄순서로 정렬하였습니다.

my_sort함수에서는 리스트의 맨 앞을 쉽게 추출하기 위하여 collections의 deque를 사용하였습니다.
while문을 통하여 q가 빌때까지 반복하는데 j, index로 q의 가장 첫번째 요소를 추출하고
나머지 q에 j보다 큰 값이 있는지, 즉 중요도가 높은 값(문서)이 있는지 확인합니다.

만약 중요도가 높은 값(문서)이 존재한다면 현재 문서 j와 index 를 q.append()를 하여 q의 가장 마지막 부분으로 옮겨 저장합니다.
그리고 break문으로 for문을 빠져나옵니다.

break에 걸리지 않고 for문을 다 돌았다면 else문에서 result에 j와 index를 저장합니다.

이렇게 정렬된 temp를 반복하면서 location과 temp[i][1]이 동일하다면 i+1을 하여 결과를 반환합니다.

 

다음은 위의 코드를 약간 수정하여 문제를 해결한 코드입니다.

다른 사람의 풀이에서 좋은 코드가 있어서 참고하였습니다.

from collections import deque
def solution(priorities, location):
    temp = [(p, i) for i, p in enumerate(priorities)]
    temp = my_sort(temp)
    for i in range(len(temp)):
        if location == temp[i][1]:
            return i+1

def my_sort(temp):
    q =deque(temp)
    result = []
    while q:
        j, index = q.popleft()
        if any(j < value[0] for value in q):
            q.append((j, index))
        else:
            result.append((j, index))
    return result

위의 코드와 많이 유사하지만 중요도가 높은 문서를 찾는 과정에서 any와 for 키워드를 사용하여 2줄로 줄여 문제를 해결한 것을 확인할 수 있습니다.
시간복잡도는 $O(n^{2})$으로 비슷한것 같지만 가독성은 더 좋아진 것 같습니다!