[C++, Golang] 2012번: 등수 매기기

 

해당 글의 내용은 최적의 알고리즘이 아닌 글쓴이의 주관적인 풀이입니다. 더 좋은 풀이가 있다면 댓글로 피드백 부탁드립니다(__)

 

링크

---

 

https://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 이하의 자연수이다.

 

 

 

입출력 예

---

<입력 1>

5
1
5
3
1
2

 

 

<출력 1>

3

 

 

풀이

---

[Golang]

package main

import (
	"bufio"
	"fmt"
	"os"
	"sort"
)

func absoluteSubtract(a int, b int) uint64 {
	result := a - b
	if result < 0 {
		return uint64(result * -1)
	}

	return uint64(result)
}

func main() {
	r := bufio.NewReader(os.Stdin)
	var num int
	_, _ = fmt.Fscanf(r, "%d\n", &num)

	expectedList := make([]int, 0, num)
	for i := 0; i < num; i++ {
		var expected int
		_, _ = fmt.Fscanf(r, "%d\n", &expected)
		expectedList = append(expectedList, expected)
	}

	sort.Ints(expectedList)

	var result uint64
	for i := 0; i < num; i++ {
		result += absoluteSubtract(expectedList[i], i+1)
	}
	fmt.Println(result)
}

 

[C++]

#include <stdio.h>
#include <vector>
#include <algorithm>

const int MAX_NUM = 500000;

int main() 
{
	std::vector<int> expectedVector;
	int num;
    
	scanf("%d", &num);
	expectedVector.reserve(MAX_NUM);

	for (int i = 0; i < num; i++)
	{
		int expected;
		scanf("%d", &expected);
		expectedVector.push_back(expected);
	}
    
	std::sort(expectedVector.begin(), expectedVector.end());

	long long result = 0;
	for (int i = 1; i <= num; i++)
	{
		int expected = expectedVector[i-1];
		if (expected - i > 0)
		{
			result += expected - i;
		}
		else
		{
			result += i - expected;
		}
	}

	printf("%lld\n", result);
    
	return 0;
}

 

풀이 설명

---

 자신이 예상한 등수와 실제 등수의 값을 최소화시키면 된다. 그러기 위해서 학생이 예상한 등수와 실제 등수를 정렬하여 두 수의 절댓값 차를 구한 뒤 모두 더해주면 불만도 합의 최솟값이 된다.

 

주의: 학생 수가 N명일 때 예상한 등수가 N 이상 일 수 있다 -_-;; (ex. 학생이 15명인데 예상한 등수를 20위라고 입력할 수 있음)