스택/큐
일반적인 프린터는 인쇄 요청이 들어온 순서대로 인쇄합니다. 그렇기 때문에 중요한 문서가 나중에 인쇄될 수 있습니다. 이런 문제를 보완하기 위해 중요도가 높은 문서를 먼저 인쇄하는 프린터를 개발했습니다. 이 새롭게 개발한 프린터는 아래와 같은 방식으로 인쇄 작업을 수행합니다.
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
를 이용하여 조건문 매우 간단하게 나타낼 수 있음!!!