[2012] 등수 매기기

🔗 문제 링크

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

🔍 문제 설명

2007년 KOI에 N명의 학생들이 참가하였다. 경시일 전날인 예비소집일에, 모든 학생들은 자신이 N명 중에서 몇 등을 할 것인지 예상 등수를 적어서 제출하도록 하였다.

KOI 담당조교로 참가한 김진영 조교는 실수로 모든 학생의 프로그램을 날려 버렸다. 1등부터 N등까지 동석차 없이 등수를 매겨야 하는 김 조교는, 어쩔 수 없이 각 사람이 제출한 예상 등수를 바탕으로 임의로 등수를 매기기로 했다.

자신의 등수를 A등으로 예상하였는데 실제 등수가 B등이 될 경우, 이 사람의 불만도는 A와 B의 차이 (|A - B|)로 수치화할 수 있다. 당신은 N명의 사람들의 불만도의 총 합을 최소로 하면서, 학생들의 등수를 매기려고 한다.

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

⚠️ 제한사항

• 1 ≤ N ≤ 500,000

• 예상 등수는 500,000 이하의 자연수

---

🗝 풀이 (언어 : Java)

개별이 아닌, 전체 불만도의 총합만 궁금하므로 예상 등수 정보를 정렬해서 1등부터 꼴찌까지 차례대로 부여하면 된다. 총합의 크기가 int 자료형의 범위를 넘을 수 있어서 long 자료형을 사용했다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        List<Integer> expectedRankList = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            expectedRankList.add(Integer.parseInt(br.readLine()));
        }
        solution(expectedRankList);
    }

    private static void solution(List<Integer> expectedRankList) {
        Collections.sort(expectedRankList);
        Queue<Integer> queue = new LinkedList<>(IntStream.rangeClosed(1, expectedRankList.size()).boxed().collect(Collectors.toList()));
        long answer = expectedRankList.stream().mapToLong(rank -> Math.abs(rank - queue.poll())).sum();
        System.out.println(answer);
    }

}