[BaekJoon] 1380 귀걸이

1.  문제 링크

2.  문제

요약

• 교감선생님이 여학생들이 찬 귀걸이를 압수하는데 압수할 때마다 해당 여학생의 번호와 'A' 또는 'B'를 함께 적어둡니다.
• 방과후 수업 이후 귀걸이를 돌려주는데 하나를 잃어버렸습니다. 이 귀걸이의 주인인 소녀의 이름을 출력합니다.
• 입력
• 시나리오들로 구성이 되는데 시나리오 번호는 1부터 증가합니다. 각 시나리오는 아래 내용을 포함합니다.
• 첫 번째 줄에는 여학생의 수, 그 뒤에 여학생 수만큼 각 줄에 여학생들의 이름이 주어집니다.
• 이후 여학생 수 * 2 - 1개의 줄에는 여학생들의 번호와 'A' 또는 'B'가 적힌 줄이 등장하는데, 각 번호는 최대 2번씩 등장할 수 있고 처음 나온 번호는 압수당했을 때, 두 번째로 등장한 번호는 돌려받았을 때를 의미합니다.
• 여학생들의 번호는 처음 주어진 이름의 순서를 의미합니다.
• 출력: 각 시나리오 번호와 함께 잃어버린 귀걸이의 주인인 소녀의 이름을 출력합니다.

3.  소스코드

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

public class Main {
	public String getLostStudent(String[] student, ArrayList<Integer> list_num) {
		String result = "";
		int[] num_count = new int[student.length];
		for(int i = 0; i < list_num.size(); i++) {
			for(int j = 1; j <= student.length; j++) {
				if(list_num.get(i) == j) {
					num_count[j - 1]++;
					break;
				}
			}
		}
		for(int i = 0; i < num_count.length; i++) {
			if(num_count[i] != 2) {
				result = student[i];
				break;
			}
		}
		return result;
	}
	
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int num = Integer.parseInt(br.readLine());
		ArrayList<String> result = new ArrayList<String>();
		Main m = new Main();
		while(num != 0) {
			String[] student= new String[num];
			for(int i = 0; i < num; i++) {
				student[i] = br.readLine();
			}
			ArrayList<String> list_code = new ArrayList<String>();
			ArrayList<Integer> list_num = new ArrayList<Integer>();
			for(int i = 0; i < num * 2 - 1; i++) {
				String temp = br.readLine();
				StringTokenizer st = new StringTokenizer(temp);
				list_num.add(Integer.parseInt(st.nextToken()));
			}
			result.add(m.getLostStudent(student, list_num));
			num = Integer.parseInt(br.readLine());
		}
		for(int i = 0; i < result.size(); i++) {
			System.out.println((i + 1) + " " + result.get(i));
		}
	}
}

4.  접근

• 결국 잃어버린 귀걸이는 한 개이기 때문에 잃어버린 귀걸이의 주인인 소녀의 번호만 한 번 나오고 나머지 번호는 두 번씩 나올 것입니다.
• 그렇기 때문에 우리는 한 번만 나오는 번호를 찾으면 주인이 누구인지 찾을 수 있습니다.
• 학생들의 이름을 배열로 저장한 후에 한 번만 나온 번호를 찾아 학생들의 이름을 저장해놓은 배열에서 (해당 번호 - 1)에 해당하는 index 값을 시나리오 순서대로 반환하면 최종 출력을 할 수 있게 됩니다.
• 그렇기 때문에 학생 이름을 찾는 로직을 함수로 만들어주어 한 시나리오마다 해당 함수를 호출하여 순서대로 ArrayList에 저장한 뒤, 순서대로 출력하면 최종 출력과 같은 형태가 됩니다.