[Baekjoon] 1268_임시 반장 정하기

Silver V

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

---

< 임시 반장 정하기 >

 

문제 풀이 

 

같은 반이었던 학생을 찾아 같은 반이었던 사람이 가장 많은 학생을 찾는다. 중복을 제거하기 위해 HashSet을 사용한다.

 

* 한 사람과 여러 학년동안 같은 반이었더라도 같은 반이었던 학생 수를 1로 카운트해야 하므로 HashSet 사용

 

 my solution (Java)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.StringTokenizer;

public class _1268_ { // 임시 반장 정하기

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

		int n = Integer.parseInt(bf.readLine());

		int arr[][] = new int[n][5];

		for (int i = 0; i < n; i++) {
			st = new StringTokenizer(bf.readLine());
			for (int j = 0; j < 5; j++) {
				arr[i][j] = Integer.parseInt(st.nextToken());
			}
		}

		ArrayList<HashSet<Integer>> result = new ArrayList<>();
		for (int i = 0; i < n; i++) {
			result.add(new HashSet<>());
		}

		ArrayList<ArrayList<Integer>> list = new ArrayList<>();
		for (int i = 0; i < 10; i++) {
			list.add(new ArrayList<>());
		}

		for (int j = 0; j < 5; j++) {
			for (int i = 0; i < 10; i++) {
				list.get(i).clear();
			}
			for (int i = 0; i < n; i++) {
				list.get(arr[i][j]).add(i);
			}
			for (int i = 0; i < n; i++) {
				if (list.get(arr[i][j]).size() > 1) {
					for (int k = 0; k < list.get(arr[i][j]).size(); k++) {
						result.get(i).add(list.get(arr[i][j]).get(k));
					}
				}
			}
		}

		int max = 0;
		int answer = 0;
		for (int i = 0; i < n; i++) {
			if (max < result.get(i).size()) {
				max = result.get(i).size();
				answer = i;
			}
		}
		System.out.println(answer + 1);
	}
}

변수)
n : 반의 학생 수
arr : 각 학생이 속했던 반 정보
result : 각 학생과 같은 반 했던 학생 번호
list : 각 학년별로 같은 반이었던 학생 번호
max, answer : 같은 반이었던 사람이 가장 많은 학생 수, 학생 번호

 

반의 학생 수 n을 입력받는다. 각 학생의 반 정보를 입력받아 arr에 저장한다. 학년별로 학생이 속한 반을 살펴보며 같은 반이었던 학생끼리 묶어둔다. 그 후 각 학생별로 자신과 같은 반이었던 학생 번호를 저장한다. 최종 result를 살펴보며 같은 반이었던 사람이 가장 많은 학생을 찾아 출력한다.

---

---