알고리즘/알고리즘_백준

[Python][백준] 1138.한 줄로 서기

Jerry_K 2024. 7. 8. 17:24

링크🔗

https://www.acmicpc.net/problem/1138


🗒️파이썬 코드 풀이

N = int(input())
lst = list(map(int,input().split()))
ord = [i for i in range(1,N+1)]

result= []

for i in range(N-1,-1,-1):
    result.insert(lst[i],ord[i])

# 리스트 값 출력을 위해 str로 변경
str_rs = [str(result[i]) for i in range(N)]
print(' '.join(str_rs))

 

1.  2개의 리스트를 만든다.  ( lst는 입력을 저장한 리스트, ord는 1~N까지의 리스트 )

 

2. 입력받은 값들을 기준으로 1~N을 거꾸로 하나 하나 삽입 해주면 된다.

🌟 insert 함수 같은 경우, lst의 범위를 초과해도 가장 마지막 인덱스에 삽입 해준다.
ex)  입력이 2 1 1 0 인 경우

- 인덱스 0에 삽입
i = 3 일때, [4] 

- 인덱스 1에 삽입
i = 2 일때, [4,3]   

- 인덱스 1에 삽입
i = 1 일때, [4,2,3] 

- 인덱스 2에 삽입
i = 0 일때, [4,2,1,3]

 

 

📌  문제 코멘트

dfs로 풀려고 시도했는데, 실패했다 ...

거꾸로도 생각해보았는데, 이 방법은 생각하지 못했다.

 

문제 유형을 잘 기억하고, 나중에 비슷한 문제가 나올 때는 꼭 해결해보자 !

 


📚문제

N명의 사람들은 매일 아침 한 줄로 선다. 이 사람들은 자리를 마음대로 서지 못하고 오민식의 지시대로 선다.

어느 날 사람들은 오민식이 사람들이 줄 서는 위치를 기록해 놓는다는 것을 알았다. 그리고 아침에 자기가 기록해 놓은 것과 사람들이 줄을 선 위치가 맞는지 확인한다.

사람들은 자기보다 큰 사람이 왼쪽에 몇 명 있었는지만을 기억한다. N명의 사람이 있고, 사람들의 키는 1부터 N까지 모두 다르다.

각 사람들이 기억하는 정보가 주어질 때, 줄을 어떻게 서야 하는지 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 사람의 수 N이 주어진다. N은 10보다 작거나 같은 자연수이다. 둘째 줄에는 키가 1인 사람부터 차례대로 자기보다 키가 큰 사람이 왼쪽에 몇 명이 있었는지 주어진다. i번째 수는 0보다 크거나 같고, N-i보다 작거나 같다. i는 0부터 시작한다.

출력

첫째 줄에 줄을 선 순서대로 키를 출력한다.

예제 입력 1 복사

4
2 1 1 0

예제 출력 1 복사

4 2 1 3

예제 입력 2 복사

5
0 0 0 0 0

예제 출력 2 복사

1 2 3 4 5

예제 입력 3 복사

6
5 4 3 2 1 0

예제 출력 3 복사

6 5 4 3 2 1

예제 입력 4 복사

7
6 1 1 1 2 0 0

예제 출력 4 복사