[c++] BOJ 1417번 :: 국회의원 선거

국회의원 선거

문제

다솜이는 사람의 마음을 읽을 수 있는 기계를 가지고 있다. 다솜이는 이 기계를 이용해서 2008년 4월 9일 국회의원 선거를 조작하려고 한다.

다솜이의 기계는 각 사람들이 누구를 찍을 지 미리 읽을 수 있다. 어떤 사람이 누구를 찍을 지 정했으면, 반드시 선거때 그 사람을 찍는다.

현재 형택구에 나온 국회의원 후보는 N명이다. 다솜이는 이 기계를 이용해서 그 마을의 주민 M명의 마음을 모두 읽었다.

다솜이는 기호 1번이다. 다솜이는 사람들의 마음을 읽어서 자신을 찍지 않으려는 사람을 돈으로 매수해서 국회의원에 당선이 되게 하려고 한다. 다른 모든 사람의 득표수 보다 많은 득표수를 가질 때, 그 사람이 국회의원에 당선된다.

예를 들어서, 마음을 읽은 결과 기호 1번이 5표, 기호 2번이 7표, 기호 3번이 7표 라고 한다면, 다솜이는 2번 후보를 찍으려고 하던 사람 1명과, 3번 후보를 찍으려고 하던 사람 1명을 돈으로 매수하면, 국회의원에 당선이 된다.

돈으로 매수한 사람은 반드시 다솜이를 찍는다고 가정한다.

다솜이가 매수해야하는 사람의 최솟값을 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 후보의 수 N이 주어진다. 둘째 줄부터 차례대로 기호 1번을 찍으려고 하는 사람의 수, 기호 2번을 찍으려고 하는 수, 이렇게 총 N개의 줄에 걸쳐 입력이 들어온다. N은 1,000보다 작거나 같은 자연수이고, 득표수는 1,000보다 작거나 같은 자연수이다.

출력

첫째 줄에 다솜이가 매수해야 하는 사람의 최솟값을 출력한다.

알고리즘 분류

[브루트 포스](https://www.acmicpc.net/problem/tag/브루트 포스)

나의 답

생각의 흐름

1. 하나씩 줄여보자

   시간복잡도 상 가능할 것 같지만 다른 방법을 한 번 찾아보았다.

2. (가장 큰 수 - 다솜이 사람 / 2) 반복하기

   반례가 나와서 실패

코드

brute force

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;
int N;
vector<int> num_arr;
int dasom;

int howManyPeople() {
    //O(n*n)
    /* 답 구하기 */
    if (num_arr.size() == 0) {
        return 0;
    }

    int answer = 0;
    int tmp;
    while (num_arr.size() != 0) {
        sort(num_arr.begin(), num_arr.end());
        if (num_arr[num_arr.size() - 1] < dasom) // 제일 큰 수보다 dasom이 크면
            break;
        num_arr[num_arr.size() - 1]--; dasom++; answer++;
        int size = num_arr.size();
        for (int i = size - 1; i >= 0; i--) {
            if (num_arr[i]<dasom)
                num_arr.erase(num_arr.begin() + i);
        }
    }

    return answer;
}

int main() {
    cin >> N;
    cin >> dasom;

    for (int i = 0; i < N - 1; i++) {
        int tmp;
        cin >> tmp;
        if (dasom <= tmp)
            num_arr.push_back(tmp);
    }

    cout << howManyPeople();


}

중간에 sort도 직접 구현하고, STL의 여러가지 컨테이너 multiset, set 등을 이용해보기도 하였으나 결국 가장 기본적인 방법으로 돌아왔다.

계속 런타임에러가 나서 의아했는데, 간단한 코드라서 함수를 구현하지 않은 것이 문제였다. 함수가 없으니 다솜이를 지지하는 사람이 가장 많을 때 return 1;을 했더니 main 함수에서 런타임에러 처리해서 계속 런타임에러가 나는 것이었다. 후... 다음에는 조심하고 삽질하지 말자 ^ㅠ^

다른 답

우선순위 큐를 이용한 풀이방법을 봤는데, 굉장히 편해보인다! 정렬을 자동으로 해주는 컨테이너를 이용하는 것이 좋을 것 같다.