알고리즘 챌린지 1일차

문제

문제: 백준알고리즘 1764번 듣보잡

풀이

• 듣도 못한 사람의 명단을 먼저 set에 넣는다.
• 보도 못한 사람이 듣도 못한 사람의 명단에 포함되어 있는지 확인한다.
• 포함되어 있으면 이 사람을(듣도 보도 못한 사람)을 ArrayList에 넣는다.
• 사전순으로 정렬한 뒤 출력한다.

코드

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] inputs = br.readLine().split(" ");
        int n = Integer.parseInt(inputs[0]);
        int m = Integer.parseInt(inputs[1]);

        HashSet<String> set = new HashSet<>(); 
        for (int i = 0; i < n; i++) {
            set.add(br.readLine());
        }

        ArrayList<String> result = new ArrayList<>();
        for (int i = 0; i < m; i++) {
            String tmp = br.readLine();
            if(set.contains(tmp)){
                result.add(tmp);
            }
        }

        Collections.sort(result);

        // print result
        System.out.println(result.size());
        for (String s : result) {
            System.out.println(s);
        }
    }
}

정리

처음에 ArrayList로 풀려고 시도했지만 시간초과가 계속나서 실패했다. 그래서 고민중에 HashSet을 생각해냈고 ArrayList대신 HashSet에 넣으니까 성공했다.

그 이유는 HashSet의 contains()는 O(1), ArrayList의 contains()는 O(n)이기 때문이다. HashSet은 map을 기반으로 구현되고, ArrayList는 indexOf()를 사용해서 contain여부를 결정한다고 한다.

효율성을 요구하는 문제를 만났을 때에는 contains()를 쓸 때 HashSet을 사용하자!