[백준] 1417번 국회의원 선거 (java,c++)

반응형
반응형

문제

다솜이는 사람의 마음을 읽을 수 있는 기계를 가지고 있다. 다솜이는 이 기계를 이용해서 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보다 작거나 같은 자연수이다.

출력

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

 

이 문제는 쉬운데 모르면 틀려야 하는 문제다.

 

핵심 키워드는 정렬과 마지막이다.

 

이 문제는 완탐, 즉 부루트포스로 푸는 문제다.

부루트 포스는 끝날때까지 뭔가가 계속 탐색,혹은 변화하는 그런 알고리즘이다. 쉽게 얘기하면

모든 것을 한다고 생각하면 쉽다.

 

근데 이 문제가 까다로운 이유는 정렬과 마지막이라는 것 때문이다. 

물론, 이 두개가 아니여도 푸는 방법이야 많겠지만...

정렬을 하게 되면 가장 큰 수가 맨 앞혹은 맨 뒤로 가게 된다. 이 성질을 이용하면 쉽게 문제를 해결 할 수 있다.

쉽게 생각해서 오름차순이라고 생각해보자

그러면 가장 큰수가 다솜이보다 작다는 이야기는 나머지는 볼 필요없이 다솜이보다 작다는 이야기가 된다.

즉, 가장 큰 놈만 확인하는게 이 문제의 핵심이다.

 

근데 나는 계속 전부 비교하려고 했으니... 당연히 찾을 수 없었던 것 같다.

뭐 브루트포스라는게 전부 비교하는 건 맞는데

정렬해서 문제를 해결할 생각은 못해봤다.

다음 에는 조금더 신경써서 문제를 해결해 보자. 

 

자바 코드 : 

package _2020_06_03.D;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main {

	static int go(int d, int[] arr, int count) {
		Arrays.sort(arr);

		int last = arr.length - 1;
		if (d > arr[last]) {
			return count;
		}

		arr[last]--;
		return go(d + 1, arr, count + 1);
	}

	public static void main(String[] args) throws NumberFormatException, IOException {
		
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

		int n = Integer.parseInt(br.readLine());
		int d = Integer.parseInt(br.readLine());

		int[] arr = new int[n - 1];

		for (int i = 0; i < arr.length; i++) {
			arr[i] = Integer.parseInt(br.readLine());
		}

		if (n == 1) {
			System.out.println(0);
			return;
		}

		System.out.println(go(d, arr, 0));

	}

}

 

부르트 포스는 푸는 방법은 2가지가 있다. 하나는 반복문을 이용하는 방법이고 

두 번째는 재귀를 이용하는 방법이다. 사실 다른 블로그를 참조했지만 블로그가 반복문을 사용했기 때문에 

나는 재귀를 사용했다.

또 다솜이가 마지막보다 크면 다솜이가 매수할 사람들의 수가 나온다.

만약, 그게 아니라면 가장 당선 될 확률인 사람을 뽑을 사람 한명을 매수해오면 된다.

그리고 그값을 다시 재귀를 넣어 주면 완성이닷!

 

마지막으로 n == 1을 하는 이유는 애초에 배열은 0이 되면 에러를 발생하기 때문에 어쩔 수 없었다.

 

여담이지만 bufferdReader는 30ms정도 빨라진것 같다.

근데... 메모리는 암울하다. ㅜㅜ

 

c++ 코드 :

#include <bits/stdc++.h>
using namespace std;
int candidate[1001];
int n;
int d;

int main(void) {
  cin >> n;

  cin >> d;

  for(int i = 0; i<n-1;i++) {
    cin >> candidate[i];
  }

  int len = n-1;
  int count = 0;
  while(true) {
    int max = 0;
    int index = 0;  
    for(int i = 0;i<len;i++) {
      if (max < candidate[i]) {
        max = candidate[i];
        index = i;
      }
    }
    
    // 기저 조건
    if (d > max) {
      cout << count << "\n";
      break;
    }

    d++;
    candidate[index]--;
    count++;
  }
} 

솔직히 재귀를 이용하고 싶었지만... c++은 메서드에 배열을 넣는 법을 몰라서 그냥 while(true)로 했다.

이 문제의 핵심이 가장 큰놈만 조지는게 이 문제의 핵심인데...

그것을 전적으로 잘 보여주는 예이다.

이번에는 정렬을 사용하지 않고 문제를 풀었다.

정렬을 이용하지 않으면 최댓값을 구해야하는데...

문제는 최댓값을 구하는건 아무 문제가 되지 않지만...

 

해당 후보자의 득표수를 강제로 줄여야 하기 때문에 index가 필요하다. 그래서 인덱스로 작성해줘야 한다.

 

잠깐더 생각해 봤는데... 메서드에 배열이 아니라 일반 변수를 넣는 방법도 있지 않을까? 

그건 좀더 생각해봐야될 과제인듯 싶다.

 

마지막으로 다솜이의 득표가 얼마나 증가되었는지 알수 있는 방법은 2가지다.

하나는 바뀐 값에서 원래 값을 빼주면 되고... 

나머지하는 지금처럼 해주면 된다.

 

#include	<iostream>
#include	<queue>

using namespace	std;

int	main(void)
{
	cin.tie(NULL);
	cin.sync_with_stdio(false);
	
	int					N,dasom,cnt;
	priority_queue<int>	candidates;
	
	cin>>N>>dasom;
	
	for(int i=1;i<N;i++)
	{
		int	n;
		
		cin>>n;
		candidates.push(n);
	}
	
	for(cnt=0;!candidates.empty()&&candidates.top()>=dasom;cnt++)
	{
		int	n;
		
		n = candidates.top();
		candidates.pop();
		
		n--;dasom++;
		candidates.push(n);
	}
	
	cout<<cnt<<'\n';
	
	return	0;
}

이 분은 조금 특이하게 푸셨는데...

큐를 이용해서 푸셨다.

이런 방법도 있구나.. ㄷㄷ  조금 깨지는게 있는데...

자바랑 c++동시에 있으면 안되는건가... ㅜㅜ

반응형

'알고리즘 > 백준' 카테고리의 다른 글

[백준] 2293번 동전 1  (0) 2020.06.04
[백준] 1343번 폴리오미노(java,c++)  (0) 2020.06.03
[백준] 14891번 톱니바퀴  (0) 2020.06.02
[백준] 14912번 숫자 빈도수  (0) 2020.05.31
[백준] 17388번 와글와글 숭고한  (0) 2020.05.30

댓글

Designed by JB FACTORY