[백준 1380/자바(java)] 귀걸이

백준(baekjoon) 1380 귀걸이

1380번: 귀걸이

문제

파스칼 고등학교에 다니는 많은 여학생들은 규정에 없는 귀걸이를 착용한 채 도망다닙니다. Sneddon 교감선생님은 흔들거리는 긴 빨간 귀걸이들을 볼때마다 압수합니다.

교감선생님은 귀걸이를 압수당한 여학생들을 숫자를 매겨 리스트를 작성하고 있습니다. 그리고 압수한 귀걸이 뒤쪽에 여학생 번호와 마음대로 선택한 'A' 또는 'B'를 함께 적어두었습니다.

모든 정규 일과와 방과후 수업의 감금이 끝나면, 여학생들은 교감선생님을 찾아와 귀걸이를 돌려받습니다. 불행하게도 어느 날, 교감선생님은 귀걸이가 든 봉투를 잃어버렸고, 하나를 끝내 찾지 못했습니다.

귀걸이를 받지 못해 화난 소녀의 이름을 교감선생님께 알려주세요.

입력

입력은 번호를 가진 시나리오들로 구성됩니다. 시나리오 번호는 1부터 순서대로 증가하고, 각 시나리오는 아래의 내용을 포함합니다.

• 한 줄에 귀걸이를 압수당한 여학생의 수, n (1 ≤ n ≤ 100)이 주어집니다.
• 다음 n줄에 걸쳐 여학생들의 이름(최대 60자)이 주어집니다.
• 다음 2n − 1줄에 여학생 번호와 'A' 또는 'B'가 공백을 사이에 두고 주어집니다. 번호는 교감선생님의 여학생 이름 리스트와 순서가 일치합니다. 즉, 1은 첫 번째로 압수당한 여학생입니다. 여학생 번호는 최대 2번 등장하며, 두 번째로 등장할 때는 첫 번째 경우와 다른 'A' 또는 'B'가 뒤에 적힙니다.
  번호가 처음 등장하는 것은 압수되었음을, 두 번째로 등장하는 것은 돌려받았음을 의미합니다.

'0'을 마지막 줄로 하여 입력이 종료됩니다. '0'은 처리하지 않습니다.

출력

시나리오 번호와 귀걸이를 돌려받지 못한 여학생의 이름을 공백으로 구분하여 한 줄씩 출력하십시오.

 

풀이과정

처음에 HashMap을 사용하여 풀이를 시도하여서 출력은 성공하였지만, 제출 결과 오답으로 나왔고, 코드 또한 너무 복잡하였다. 리팩토링하면서 더 나은 방법을 찾다가, 굳이 Key, Value 형식을 쓸 필요가 없다는 결론을 내렸고, HashMap 대신 ArrayList를 사용하기로 하였다.

 

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int scenario = 0; // 현재 시나리오 번호

        while (true) {
            int n = Integer.parseInt(br.readLine());
            if (n == 0) {
                return; // 입력이 0일 경우 종료
            } else {
                scenario++;
            }

            ArrayList<String> girls = new ArrayList<>(); // 여학생 이름을 저장할 리스트

            // 여학생들의 이름 입력
            for (int i = 0; i < n; i++) {
                girls.add(br.readLine());
            }

            ArrayList<Integer> lostEarrings = new ArrayList<>(); // 귀걸이를 잃어버린 학생 번호를 저장할 리스트

            // 여학생들의 귀걸이 상태 입력
            for (int i = 0; i < n * 2 - 1; i++) {
                StringTokenizer st = new StringTokenizer(br.readLine());
                int num = Integer.parseInt(st.nextToken());

                if (lostEarrings.contains(num)) {
                    lostEarrings.remove((Integer) num); // 귀걸이를 되돌려받은 경우
                } else {
                    lostEarrings.add(num); // 처음 잃어버린 경우
                }
            }

            int lostGirlIndex = lostEarrings.get(0); // 귀걸이를 잃어버린 학생의 번호

            // 결과 출력
            System.out.println(scenario + " " + girls.get(lostGirlIndex - 1));
        }
    }
}