[백준] 국회의원 선거(C++)

comomo·2024년 4월 11일

코딩연습

목록 보기

17/28

문제

국회의원 선거
어제가 문제 제목이였는데 사전투표기간에 투표를 하지 못해 투표를 못했는데 대신 문제라도....!

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

다솜이의 기계는 각 사람들이 누구를 찍을 지 미리 읽을 수 있다. 어떤 사람이 누구를 찍을 지 정했으면, 반드시 선거때 그 사람을 찍는다.
현재 형택구에 나온 국회의원 후보는 N명이다. 다솜이는 이 기계를 이용해서 그 마을의 주민 M명의 마음을 모두 읽었다.

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

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

돈으로 매수한 사람은 반드시 다솜이를 찍는다고 가정한다.
다솜이가 매수해야하는 사람의 최솟값을 출력하는 프로그램을 작성하시오.

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

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

우선순위 큐

우선순위 큐는 우선순위의 개념을 큐에 도입한 자료구조이다.
각각의 데이터는 우선순위를 가지고 있으며 우선순위가 높은 데이터가 먼저 나가게 된다.

우선순위 큐의 구현 방법에는 배열, 연결리스트, 히프 등을 이용하는 방법들이 있는데 이 문제에서는 히프를 사용할것이다.

히프

여러개의 값들 중에서 가장 큰 값이나 가장 작은 값을 빠르게 찾아내도록 만들어진 자료구조

문제분석

선거에 나가는 후보들의 득표수를 확인하고 다솜이가 당선이 되기 위해서 메수해야하는 사람의 최소인원을 구하는 것이 목표이다.
일단 문제가 우선순위 큐로 분류되어 있기때문에 우선순위 큐를 이용하여 푸는 문제일 것이다.

max heap를 이용하여 우선순위 큐를 구현하였으며 문제의 해결방식은 다음순서와 같다.

다솜이의 득표수를 제외한 후보들의 득표수로 heap을 만든다.
루트 노드의 값과 다솜이의 득표수를 비교한다.
다솜이의 득표수가 더 작다면 루트노드는 1을 감소시키고 다솜이의 득표수는 1증가시킨다.
heap를 재배치한다.
다솜이의 득표수가 루트노드의 값보다 커질때 까지 반복한다.

코드

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;


int main() {
	vector<int>v;
	int n, num, answer = 0;
	cin >> n;
	for (int i = 0; i < n; i++)
	{
		cin >> num;
		v.push_back(num);
	}
	if (n == 1) {
		cout << 0;
		return 0;
	}

	make_heap(v.begin()+1, v.end());
	
	while (v[0] <= v[1])
	{
		v[1]--;
		v[0]++;
		answer++;
		make_heap(v.begin()+1, v.end());
	}
	cout << answer;
}