국회의원 선거(백준 1417번) [JAVA]

 

문제

문제 링크 : www.acmicpc.net/problem/1417

1417번: 국회의원 선거

시간 제한	메모리 제한	제출	정답	맞은 사람	정답 비율
2초	128 MB	1236	366	322	30.696%

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

출력

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

 

풀이

사실 1000개의 배열중 가장 큰 값을 찾고, -1씩 빼가면서 단순하게 하려고 했었으나 시간초과를 맞았다.

그래서 생각한게, 배열중 어차피 기호 1번(n)보다 큰 수들만 생각하면 되므로, 걔네를 따로 뽑아서 ArrayList 인 ranker에 넣었다.

Ranker에 아무도 없으면 1번이 제일 큰 것이므로 바로 종료하고 --> 이거 안해서 런타임에러 8번 먹었다.

ranker를 오름차순으로 정렬한 다음에 맨 마지막 (제일 큰애)를 비교해서 vip(기호 1번)이 더크면 break; 아니면 표 하나 매수하고

for문으로 Ranker에서 1번보다 작은애를 삭제한다음에 다시 정렬하며 반복한다.

정렬의 복잡도를 nlgn이라 놓으면, 시간 복잡도는 nlgn * n(for문)이 n번 (정답이 나올때까지) 돌으므로 $O(n^3lgn)$인가 싶다.

 

코드

import java.util.ArrayList;
import java.util.Collections;

public class Main {
    public static StringBuilder sb = new StringBuilder();
    public static java.io.BufferedReader br = new java.io.BufferedReader(new java.io.InputStreamReader(System.in));
    static int n, ans, vip;
    static int numbers[];
    static ArrayList<Integer> ranker;
    public static void main(String[] args)throws Exception {
    	
    		int n = Integer.parseInt(br.readLine());
    		ranker = new ArrayList<Integer>();
    		numbers = new int[n];
    		vip = Integer.parseInt(br.readLine());
    		
    		for(int i=0; i <n-1; i++) {
    			int e = Integer.parseInt(br.readLine());
    			if(vip <= e) {
    				ranker.add(e);
    			}
    		}
    		ans = 0;
    		Manipulation();
    		System.out.println(ans);
    }
    public static void Manipulation() {

    	while(!ranker.isEmpty()) {
    		//젤큰애가 맨뒤에
        	Collections.sort(ranker);
        	int size = ranker.size()-1;
        	int vote = ranker.get(size);        	
        	if(vip > vote) break;
    		vip++; ans++;
    		ranker.set(size, --vote);	
    		
    		for(int i = size; i >= 0; --i) {
    			if(vip > ranker.get(i)) ranker.remove(i);
    		}
    	} 	
    }
}

 

결과