[Baekjoon] 6160_Election Time

문제(출처: https://www.acmicpc.net/problem/6160)

---

< Election Time >

 

문제 풀이 

 

1라운드) A 값을 기준으로 내림차순 한 뒤 K개를 뽑아 저장한다.

2라운드) B값을 기준으로 내림차순 한 뒤 0번째 값의 index를 출력한다.

 

* 1라운드 진행 후 K개를 따로 뽑아 저장하지 않고 K개를 탐색하여 단순 비교로도 답을 구할 수 있다.

 

 my solution (Java)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.Comparator;
import java.util.StringTokenizer;
public class _6160_ { // Election Time
	static class Info {
		private int idx;
		private int A;
		private int B;
		public Info(int idx, int A, int B) {
			this.idx = idx;
			this.A = A;
			this.B = B;
		}
	}
	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(bf.readLine());
		int N = Integer.parseInt(st.nextToken());
		int K = Integer.parseInt(st.nextToken());
		Info[] voteA = new Info[N];
		for (int i = 0; i < N; i++) {
			st = new StringTokenizer(bf.readLine());
			voteA[i] = new Info((i + 1), Integer.parseInt(st.nextToken()), Integer.parseInt(st.nextToken()));
		}
		Arrays.sort(voteA, new Comparator<Info>() {
			@Override
			public int compare(Info o1, Info o2) {
				return o2.A - o1.A;
			}
		});
		Info[] voteB = new Info[K];
		for (int i = 0; i < K; i++) {
			voteB[i] = voteA[i];
		}
		Arrays.sort(voteB, new Comparator<Info>() {
			@Override
			public int compare(Info o1, Info o2) {
				return o2.B - o1.B;
			}
		});
		System.out.println(voteB[0].idx);
	}
}

변수)
N, K : 사람 수, 1라운드에서 뽑을 인원수
voteA, voteB : 1라운드, 2라운드 투표 정보

 

Info

index, 첫 번째 라운드 투표 결과, 두 번째 라운드 투표 결과를 변수로 가짐

 

Main

N, K를 입력받는다. N만큼 투표 정보를 입력받아 voteA 배열에 저장한다. voteA 배열을 첫 번째 라운드 투표 결과 기준 내림차순으로 정렬한 후 K개만큼 voteB 배열에 저장한다. voteB 배열을 두 번째 라운드 투표 결과 기준 내림차순으로 정렬한 후 0번째 값의 index를 출력한다.

---

---