(Python/🥈4)백준 알고리즘 1015번 : 수열정렬

 

문제 바로가기

수열정렬

 

문제:

P[0], P[1], ...., P[N-1]은 0부터 N-1까지(포함)의 수를 한 번씩 포함하고 있는 수열이다. 수열 P를 길이가 N인 배열 A에 적용하면 길이가 N인 배열 B가 된다. 적용하는 방법은 B[P[i]] = A[i]이다. 배열 A가 주어졌을 때, 수열 P를 적용한 결과가 비내림차순이 되는 수열을 찾는 프로그램을 작성하시오. 비내림차순이란, 각각의 원소가 바로 앞에 있는 원소보다 크거나 같을 경우를 말한다. 만약 그러한 수열이 여러개라면 사전순으로 앞서는 것을 출력한다.

입력:

첫째 줄에 배열 A의 크기 N이 주어진다. 둘째 줄에는 배열 A의 원소가 0번부터 차례대로 주어진다. N은 50보다 작거나 같은 자연수이고, 배열의 원소는 1,000보다 작거나 같은 자연수이다.

출력:

첫째 줄에 비내림차순으로 만드는 수열 P를 출력한다.

 

풀이:

import sys
sys.stdin = open('/Users/song/Desktop/Python/Python/h.txt', 'r')

n = int(input())
lst = list(map(int,input().split()))
sorted_list = sorted(lst)
res = []
for i in range(n):
    idx = sorted_list.index(lst[i])
    res.append(idx)
    sorted_list[idx] = False
print(res)

 

입력값인 lst와 정렬된 lst를 비교해

lst에 있는 값이 정렬된 lst에 몇번 인덱스에 있는지 이다.

False 로 바꿔준 이유는 중복된 숫자를 방문 처리해주는것이다.

 

2 1 3 1 일때

앞에 있는 1을 찾고 나서, 그다음에 1을 찾을때 같은 위치이기때문에 그것을 방지하기 위해

앞에 있는 1을 False로 방문처리해야 다음 포문일때 뒤에 1에 인덱스를 가져온다.

 

쉽지 않았다.