[백준 1764] 듣보잡 (Java) - 정렬

https://www.acmicpc.net/problem/1764

듣보잡

문제

김진영이 듣도 못한 사람의 명단과, 보도 못한 사람의 명단이 주어질 때, 듣도 보도 못한 사람의 명단을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 듣도 못한 사람의 수 N, 보도 못한 사람의 수 M이 주어진다. 이어서 둘째 줄부터 N개의 줄에 걸쳐 듣도 못한 사람의 이름과, N+2째 줄부터 보도 못한 사람의 이름이 순서대로 주어진다. 이름은 띄어쓰기 없이 알파벳 소문자로만 이루어지며, 그 길이는 20 이하이다. N, M은 500,000 이하의 자연수이다.

듣도 못한 사람의 명단에는 중복되는 이름이 없으며, 보도 못한 사람의 명단도 마찬가지이다.

출력

듣보잡의 수와 그 명단을 사전순으로 출력한다.

예제 입력 1 복사

3 4
ohhenrie
charlie
baesangwook
obama
baesangwook
ohhenrie
clinton

예제 출력 1 복사

2
baesangwook
ohhenrie

---

풀이

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

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

        int N = Integer.parseInt(st.nextToken()); // 듣도 못한 사람 수
        int M = Integer.parseInt(st.nextToken()); // 보도 못한 사람 수

        ArrayList<String> answer = new ArrayList<>(); // 듣도보도 못한 사람
        HashSet<String> everHeard = new HashSet<>(); // 듣도 못한 사람
        for (int i=0; i<N; i++){
            everHeard.add(br.readLine());
        }

        // 보도 못한 사람
        for (int i=0; i<M; i++){
            String everSeen = br.readLine();
            if (everHeard.contains(everSeen)){
                answer.add(everSeen);
            }
        }

        // 듣도보도 못한 사람 리스트 사전순 정렬
        answer.sort(Comparator.naturalOrder());

        StringBuilder sb = new StringBuilder();

        // 듣보잡 명수
        sb.append(answer.size() + "\n");

        // 명단 출력
        for (String everHeardAndSeen : answer){
            sb.append(everHeardAndSeen + "\n");
        }

        System.out.println(sb);
    }
}

• 정렬과 HashSet을 활용한 문제
• 듣도 못한 사람과 보도 못한사람 명단 중 두 명단에 포함되는 사람들을 찾아서 오름차순으로 정렬한다.
• 두 명단에 중복되는 사람을 찾기 위해서는 HashSet을 사용할 수 있다.
• ArrayList를 선언해서 List.contains()를 활용해도 상관을 없을 것 같지만 Hash가 O(1)로 빠른 검색속도를 보장하므로 Hash를 사용했다.
• HashSet과 HashMap중 Set을 사용한 이유는 굳이 key, value의 자료구조는 필요하지 않기때문에 Set을 사용했다.

 

HashSet 사용

1. 듣도 못한 사람 N명을 HashSet에 저장
2. 보도 못한 사람 M명을 입력 받으면서:
   • 이미 HashSet에 있으면 → 듣보잡 명단에 추가
3. 듣보잡 명단을 사전순 정렬
4. 개수와 명단 출력