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

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

1015번: 수열 정렬

Problem

• 배열 A가 주어졌을 때, 수열 P를 적용한 결과가 비내림차순이 되는 수열을 찾는 프로그램을 작성하시오. 

조건

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

SOL

Custom sorting을 하는 문제이다. 문제를 말로 설명하면 , 엄청 복잡해지는 신기한 문제이다.

B[P[0]] = A[0] ,B[P[1]] = A[1] ,B[P[2]] =A[2] ... 로, 수열 P를 적용한 결과가 비 내림차순이 되는 수열을 찾는 프로그램을 작성하라고 했기 때문에, 우선 배열 B가 비내림차순을 이루어야한다.

 

따라서, B는 A의 값에 따라서, 비내림차순(오름차순)으로 Sorting되어야한다.

이때, 우리가 출력하는건 P이므로, 정렬된 값의 원래 A의 idx에 현재 B의 인덱스를 넣어주면 되는 문제이다.

 

예를들어, A= [2,3,1] 이라면, (A[i],i)의 형태로 B에 잠시 배열을 저장한다. B = [ (2,0) ,(3,1),(1,2) ]가 될거고, 값을 기준으로 sort하면, B = [ (1,2),(2,0),(3,1)]이 될거다. 즉, B[P[2]] = 1 , B[P[0]] =2 , B[P[1]] =3 과 같이 된거다.

따라서, P[2] = 0, P[0] =1, P[1] =2에 대응되어야 한다. 구현 하려면, P[B[i][1]] =i로 대응시켜주면된다!

 

#수열을 정해보아요! 
# index mapping?

N = int(input())
A = list(map(int,input().split()))
tmp = []
P=[0]*N
#B[P[i]] = A[i]
for i in range(N):
    tmp.append((A[i],i))

tmp.sort()

#tmp[][0] = A[i],
#tmp[][1] = idx
for i in range(N):
    P[tmp[i][1]] = i

print(*P)