스택/큐
일반적인 프린터는 인쇄 요청이 들어온 순서대로 인쇄합니다. 그렇기 때문에 중요한 문서가 나중에 인쇄될 수 있습니다. 이런 문제를 보완하기 위해 중요도가 높은 문서를 먼저 인쇄하는 프린터를 개발했습니다. 이 새롭게 개발한 프린터는 아래와 같은 방식으로 인쇄 작업을 수행합니다.
1. 인쇄 대기목록의 가장 앞에 있는 문서(J)를 대기목록에서 꺼냅니다.
2. 나머지 인쇄 대기목록에서 J보다 중요도가 높은 문서가 한 개라도 존재하면 J를 대기목록의 가장 마지막에 넣습니다.
3. 그렇지 않으면 J를 인쇄합니다.
예를 들어, 4개의 문서(A, B, C, D)가 순서대로 인쇄 대기목록에 있고 중요도가 2 1 3 2 라면 C D A B 순으로 인쇄하게 됩니다.
내가 인쇄를 요청한 문서가 몇 번째로 인쇄되는지 알고 싶습니다. 위의 예에서 C는 1번째로, A는 3번째로 인쇄됩니다.
현재 대기목록에 있는 문서의 중요도가 순서대로 담긴 배열 priorities와 내가 인쇄를 요청한 문서가 현재 대기목록의 어떤 위치에 있는지를 알려주는 location이 매개변수로 주어질 때, 내가 인쇄를 요청한 문서가 몇 번째로 인쇄되는지 return 하도록 solution 함수를 작성해주세요.
def solution(priorities, location):
answer = 0
name = [i for i in range(len(priorities))]
line = [i for i in range(len(priorities))]
for i in range(len(name)) :
for j in range(i+1, len(name)) :
if priorities[i] < priorities[j] :
line.append(name[i])
line.remove(name[i])
answer = line.index(location) + 1
return answer
처음엔 key 를 문서이름 (0, 1, 2, 3, ...), value를 우선순위 값으로 하는 dict로 문제를 풀려다가 dict에는 순서 및 인덱스 개념이 없어 적합하지 않다고 판단되어 다른 방법으로 다시 시도했다,, 그래서 위처럼 작성했다가 오류남.. 이유인 즉슨, 이 for문 반복문은 한 번만 실행되기 때문에 한 번의 루프가 끝났을 때 우선순위가 아직 정렬되지 않은 문서들이 존재하기 때문,, ㅜ 그래서 결국 구글링으로 답을 컨닝했다 ㅜ
def solution(priorities, location):
answer = 0
array1 = [c for c in range(len(priorities))] # index 위치 저장
array2 = priorities.copy() # 값 저장 (출력되는 값)
i = 0
while True:
if array2[i] < max(array2[i+1:]):
array1.append(array1.pop(i))
array2.append(array2.pop(i))
else:
i += 1
if array2 == sorted(array2, reverse=True):
break
return array1.index(location) + 1
내가 시도했던 거랑 비슷한 것 같다..!(고 우겨본다.) 우선순위 리스트를 우선순위 규칙대로 정렬하면서 우선순위가 낮을 경우 array1과 array2에서 해당 인덱스(맨 앞에 있는 원소겠죠?)를 삭제하고 append하여 맨 마지막 순서로 다시 넣어준다. while문을 이용하여 이런식으로 무한반복하고 우선순위가 다 정렬됐을 때 break 한다.
이 풀이에서 배워갈 것
while 반복문을 어찌해야 되나 싶었는데 i += 1을 하고 종료 조건으로 우선순위가 내림차순으로 정렬됐는지를 설정하면 된다.append랑 remove를 각각 작성하는 대신 pop을 이용해 append를 해주면 한 줄로 해결 가능하다.remove( )
- 리스트에서 해당 원소 삭제
- 중복되는 원소가 있을 경우 맨 앞에 있는 원소 삭제
pop( )
- 리스트에서 해당 인덱스 삭제
- 해당 인덱스의 원소 반환a = [1, 2, 3, 4, 5, 6, 7, 8, 9] a.pop(3) >>4 print(a) >>[1, 2, 3, 5, 6, 7, 8, 9]
def solution(priorities, location):
queue = [(i,p) for i,p in enumerate(priorities)]
answer = 0
while True:
cur = queue.pop(0)
if any(cur[1] < q[1] for q in queue):
queue.append(cur)
else:
answer += 1
if cur[0] == location:
return answer
enumerate를 이용해서 문서이름과 우선순위를 같이 큐에 저장. 큐의 맨 앞에 있는 (문서, 우선순위) 쌍을 꺼내어 그 뒤에 있는 우선수위들과 비교하여 더 높은 우선순위가 존재하면 이 쌍을 큐의 맨 뒤로 보낸다. 이런 식으로 무한 반복하여 정렬이 끝나면 (else) 큐의 맨 앞 문서가 location과 같은지 비교하여 answer를 return한다.
이 풀이에서 배워갈 것
enumerate를 이용하면 문서이름을 나타내를 리스트를 따로 만들지 않아도 됨. 이게 내가 원한 index가 존재하는 dict 형태인듯any를 이용하여 조건문 매우 간단하게 나타낼 수 있음!!!