[🥈4 / 백준 1015 / 파이썬] 수열 정렬

 

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를 출력한다.

 

예제 입력 1 3 2 3 1	예제 출력 1 1 2 0
예제 입력 2 4 2 1 3 1	예제 출력 2 2 0 3 1
예제 입력 3 8 4 1 6 1 3 6 1 4	예제 출력 3 4 0 6 1 3 7 2 5

 

풀이

 

문제 이해가 조금 난해했었던 문제이다.

예제 2를 예로 들자.

 

2 1 3 1 을 오름차순으로 정렬하면 1 1 2 3 이다.

이제 2 1 3 1 각각의 숫자가 정렬한 리스트에서 몇 번째 인덱스인지 출력하면 된다.

따라서 2 0 3 1 을 출력하면 정답이 된다.

 

입력이 작기 때문에 배열에서 순간순간의 최소값이 얼마인지 찾아 기록하는 O(N^2)의 방식을 사용해도 크게 문제가 없다.

필자는 문제를 O(N) 만에 해결하기 위해 딕셔너리를 이용했다. (실제론 2N)

 

input()
arr = list(map(int,input().split()))
sorted_arr = sorted(arr) + [-1]
index, cur_num, cur_nums = 0, sorted_arr[0], []
dt = dict()
for i in sorted_arr:
    if i == cur_num:
        cur_nums.append(index)
    else:
        dt[cur_num] = cur_nums[::-1]
        cur_nums = [index]
        cur_num = i
    index+=1

ans = []
for i in arr:
    ans.append(dt[i].pop())
print(*ans)