https://www.acmicpc.net/workbook/view/8399
싸피 오픈톡방에서 받은 추천문제 푸는 중!!
https://www.acmicpc.net/problem/2605
문제 설명
점심시간이 되면 반 학생 모두가 한 줄로 줄을 서서 급식을 탄다. 그런데 매일 같이 앞자리에 앉은 학생들이 앞에 줄을 서 먼저 점심을 먹고, 뒷자리에 앉은 학생들은 뒤에 줄을 서 늦게 점심을 먹게 된다. 어떻게 하면 이러한 상황을 바꾸어 볼 수 있을까 고민하던 중 선생님이 한 가지 방법을 내 놓았다. 그 방법은 다음과 같다.
학생들이 한 줄로 줄을 선 후, 첫 번째 학생부터 차례로 번호를 뽑는다. 첫 번째로 줄을 선 학생은 무조건 0번 번호를 받아 제일 앞에 줄을 선다. 두 번째로 줄을 선 학생은 0번 또는 1번 둘 중 하나의 번호를 뽑는다. 0번을 뽑으면 그 자리에 그대로 있고, 1번을 뽑으면 바로 앞의 학생 앞으로 가서 줄을 선다. 세 번째로 줄을 선 학생은 0, 1 또는 2 중 하나의 번호를 뽑는다. 그리고 뽑은 번호만큼 앞자리로 가서 줄을 선다. 마지막에 줄을 선 학생까지 이와 같은 방식으로 뽑은 번호만큼 앞으로 가서 줄을 서게 된다. 각자 뽑은 번호는 자신이 처음에 선 순서보다는 작은 수이다.
예를 들어 5명의 학생이 줄을 서고, 첫 번째로 줄을 선 학생부터 다섯 번째로 줄을 선 학생까지 차례로 0, 1, 1, 3, 2번의 번호를 뽑았다고 하자, 첫 번째 학생부터 다섯 번째 학생까지 1부터 5로 표시하면 학생들이 줄을 선 순서는 다음과 같이 된다.
- 첫 번째 학생이 번호를 뽑은 후 : 1
- 두 번째 학생이 번호를 뽑은 후 : 2 1
- 세 번째 학생이 번호를 뽑은 후 : 2 3 1
- 네 번째 학생이 번호를 뽑은 후 : 4 2 3 1
- 다섯 번째 학생이 번호를 뽑은 후 : 4 2 5 3 1
따라서 최종적으로 학생들이 줄을 선 순서는 4, 2, 5, 3, 1이 된다.
줄을 선 학생들이 차례로 뽑은 번호가 주어질 때 학생들이 최종적으로 줄을 선 순서를 출력하는 프로그램을 작성하시오.
입력
첫째 줄에는 학생의 수가 주어지고 둘째 줄에는 줄을 선 차례대로 학생들이 뽑은 번호가 주어진다. 학생의 수가 100 이하이고, 학생들이 뽑는 번호는 0 또는 자연수이며 학생들이 뽑은 번호 사이에는 빈 칸이 하나씩 있다.
출력
학생들이 처음에 줄을 선 순서대로 1번부터 번호를 매길 때, 첫째 줄에 학생들이 최종적으로 줄을 선 순서를 그 번호로 출력한다. 학생 번호 사이에는 한 칸의 공백을 출력한다.
예제 입력
5
0 1 1 3 2
예제 출력
4 2 5 3 1
풀이
import sys
read = sys.stdin.readline
N = int(read())
A = list(map(int, read().split()))
ans = []
for i,v in enumerate(A):
ans.insert(v, i+1)
print(*reversed(ans))문제에서 주어지는 예제 그대로 풀려고 하면 코드 길이가 길게 작성된다.
예제에서는 왼쪽에 있는 숫자가 오른쪽 숫자보다 앞에 선 사람이다. 그렇다면 사람을 줄에 세울 때 뒤에서 앞에 위치시키려는 코드를 작성해야 해서 머리가 복잡해진다.
애초에 그렇게 하지 말고 리스트의 오른쪽에 계속 추가되는 속성답게 오른쪽에 있는 사람을 앞에 있는 사람으로 두어 계산한 다음 뒤집어버리면 그만이다.
ans.insert(v, i+1)사람을 그 사람이 제치는 사람 수(A의 원소) 만큼의 인덱스인 부분에 삽입하면 그만큼 사람을 제쳐서 앞에 설 수 있다.
결과
