[백준 2012번] 파이썬 - 등수 매기기

http://www.acmicpc.net/problem/2012

2012번: 등수 매기기

 

 

# 조건

• 2007년 KOI에 N명의 학생들이 참가하였다. 경시일 전날인 예비소집일에, 모든 학생들은 자신이 N명 중에서 몇 등을 할 것인지 예상 등수를 적어서 제출하도록 하였다.
• KOI 담당조교로 참가한 김진영 조교는 실수로 모든 학생의 프로그램을 날려 버렸다.
• 1등부터 N등까지 동석차 없이 등수를 매겨야 하는 김 조교는, 어쩔 수 없이 각 사람이 제출한 예상 등수를 바탕으로 임의로 등수를 매기기로 했다.
• 자신의 등수를 A등으로 예상하였는데 실제 등수가 B등이 될 경우, 이 사람의 불만도는 A와 B의 차이 (|A - B|)로 수치화할 수 있다.
• 당신은 N명의 사람들의 불만도의 총 합을 최소로 하면서, 학생들의 등수를 매기려고 한다.

각 사람의 예상 등수가 주어졌을 때, 김 조교를 도와 이러한 불만도의 합을 최소로 하는 프로그램을 작성하시오.

 

입력

첫째 줄에 자연수 N이 주어진다.
(1 ≤ N ≤ 500,000) 둘째 줄부터 N개의 줄에 걸쳐 각 사람의 예상 등수가 순서대로 주어진다. 예상 등수는 500,000 이하의 자연수이다.

 

출력

첫째 줄에 불만도의 합을 최소로 할 때, 그 불만도를 출력한다.

 

# 접근 방법

• 예상 등수를 저장 후 오름차순으로 정렬해준다.
• 오름차순 정렬이 된다면 예상 등수와 실제 등수의 차이가 최소가 되기 때문이다.
• 따라서 각 예상등수와 실제 등수에 대한 불만도를 더해주면 끝

import sys  
  
n = int(sys.stdin.readline())  
ex = [int(sys.stdin.readline()) for x in range(n)]  
# 각 사람이 예상한 등수를 오름차순으로 정렬한다.  
ex.sort()  
# 랭크를 사람의 수만큼 만든다.  
rank = [i for i in range(1, n+1)]  
# 불만도 최소합  
result = 0  
  
# 랭크에서 각 사람이 예상한 등수를 빼면 불만도에 최소를 구할수있다.  
for i in range(n):  
    result += abs(rank[i] - ex[i])  
  
print(result)

 

다른 분 코드

 

• enumerate를 이용하여 실제 등수와 현재 등수의 차이를 구해준다.
• 이 때, 실제 등수가 현재 등수가 +1 이 되는 상태이므로 -1을 해준다.

_,*l=map(int,open(0))
print(sum(abs(x-i-1)for i,x in enumerate(sorted(l))))