[Baekjoon] 9491_Politics

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

---

< Politics >

 

문제 풀이 

 

HashMap을 사용하여 각 후보자별 지지자의 이름을 구한다.

 

my solution (Java)

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.StringTokenizer;

public class _9491_ { // Politics

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		StringTokenizer st;

		String str = "";

		while (!(str = bf.readLine()).equals("0 0")) {
			st = new StringTokenizer(str);

			int n = Integer.parseInt(st.nextToken());
			int m = Integer.parseInt(st.nextToken());

			ArrayList<String> list = new ArrayList<>();
			HashMap<String, ArrayList<String>> map = new HashMap<>();

			for (int i = 0; i < n; i++) {
				String name = bf.readLine();

				list.add(name);
				map.put(name, new ArrayList<>());
			}

			for (int i = 0; i < m; i++) {
				st = new StringTokenizer(bf.readLine());

				String supporter = st.nextToken();
				String cadidate = st.nextToken();

				if (map.containsKey(cadidate)) {
					map.get(cadidate).add(supporter);
				} else {
					list.add(cadidate);
					map.put(cadidate, new ArrayList<>());
					map.get(cadidate).add(supporter);
				}
			}

			for (int i = 0; i < list.size(); i++) {
				for (int j = 0; j < map.get(list.get(i)).size(); j++) {
					bw.write(map.get(list.get(i)).get(j) + "\n");
				}
			}
		}
		bw.flush();
	}
}

변수)
n, m : 후보자의 수, 지지자의 수
list : 후보자 표기 순서
map : 후보자별 지지자
name : 후보자 이름
supporter, cadidate : 지지자의 이름, 후보자의 이름

 

0 0이 입력되지 않을 때까지 다음 과정을 반복한다.

 

1) 후보자의 수, 지지자의 수 입력

2) 후보자의 수만큼 후보자 이름을 입력받아 ArrayList와 HashMap에 추가

3) 지지자의 수만큼 지지자의 이름과 후보자의 이름을 입력받아 HashMap에 후보자가 key값으로 있다면 value에 지지자의 이름 추가, key값에 없다면 새로 추가

4) ArrayList에 있는 후보자의 이름 순으로 후보자를 지지하는 지지자의 이름 출력

---

---