for문 : 학생 수 만큼 반복하는 반복문을 작성하였다. 첫 번째 학생(i = 0)의 경우는 무조건 통과하게 되고, 두 번째 학생(i = 1)부터 본격적으로 반복문을 수행하게 되는데, for문 내의 for문으로 해당 학생이 뽑은 번호 만큼 자리 이동이 될 수 있도록 작성하였다. 예를 들어 4번째 학생(i = 3)이 뽑은 번호는 <3>인데, 이때 인덱스 3을 인덱스 2의 값으로 바꾸고,인덱스 2를 인덱스 1의 값으로 바꾸고인덱스 1을 인덱스 0의 값으로 바꾸고, for문을 나와서 인덱스 <i - 뽑은 번호> 를 학생 번호로 바꿔주면 된다.(뽑은 번호만큼 앞으로 가야하니까)
n = int(input())
num_list = list(map(int, input().split())) # 0 1 1 3 2
students = list(range(1,n+1))
for i in range(len(students)):
# i = 0 / 1 2 3 4 5
# i = 1 / 1 1 3 4 5 / 2 1 3 4 5
# i = 2 / 2 1 1 4 5 / 2 3 1 4 5
# i = 3 / 2 3 1 1 5 / 2 3 3 1 5 / 2 2 3 1 5 / 4 2 3 1 5
# i = 4 / 4 2 3 1 1 / 4 2 3 3 1 / 4 2 5 3 1(end)
if i == 0:
continue
for j in range(num_list[i]):
students[i-j] = students[i-j-1]
students[i - num_list[i]] = i + 1
for i in range(len(students)):
print(students[i], end = " ")
if i == len(students) - 1:
print()
다른 코드 및 설명
insert() 메서드를 이용해 리스트의 원하는 인덱스 지점에 값을 추가할 수 있다.
n = int(input())
arr = list(map(int, input().split()))
li = []
# 0 1 1 3 2
for i in range(n):
if i == 0:
li.insert(0, i+1)
else:
li.insert(arr[i], i+1)
li.reverse()
for i in range(len(li)):
print(li[i], end = " ")
if i == len(li) - 1:
print()
insert(위치, 값) : 리스트의 원하는 위치에 값을 추가한다.
>>> nums = [1, 2, 3]
>>> nums.insert(0, [10, 20]) # 0번째(맨앞에) 추가
[[10, 20], 1, 2, 3]
>>> nums.insert(-1, 100) # 끝에서 1번째 전에 추가
>>> print(nums)
[[10, 20], 1, 2, 100, 3] # 리스트 맨 끝에 저장되지 않음
>>> nums = [1, 2, 3]
>>> nums.insert(len(nums), 100)
[1, 2, 3, 100]
extend(리스트) : 리스트를 확장한다.
>>> nums = [1, 2, 3]
>>> nums.extend([4, 5])
[1, 2, 3, 4, 5] #리스트로 주어진 [4, 5]의 요소가 각각 추가 되었음
>>> a = [10]
>>> nums.extend(a)
[1, 2, 3, 4, 5, 10]
