백준 1764 듣보잡

Algorithm/백준

백준 1764 듣보잡

chbong 2022. 8. 25. 22:23

문제

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

입력

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

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

출력

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

package ch.baekjoon;


import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

/*
백준 1764 듣보잡
 */
public class Algorithm27 {
    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> arrayList = new ArrayList<>();

        Map<String,String> map = new HashMap<>();
        for(int i = 0 ; i < n ; i ++){
            String temp = br.readLine();
            map.put(temp, temp);
        }

        for(int i = 0; i < m ; i ++){
            String temp = br.readLine();
            if(map.containsKey(temp)){
                arrayList.add(map.get(temp));
            }
        }
        Collections.sort(arrayList);
        System.out.println(arrayList.size());
        for(String s : arrayList){
            System.out.println(s);
        }
    }
}

풀이

우선 조금 반복문을 여러개 사용하여 풀었더니 시간초과가 떳다. 백준문제를 풀때마다 시간초과는 항상 따라다니는 것 같다.

일단 중복이 없는것을 확인하였으니 map에 듣도못한 사람들을 차례로 넣는다.

후에 보도못한 사람들을 containsKey 를 통하여 boolean값으로 반환받아서 만약 참이라면 arrayList에 add한다.

출력에서 몇명의 사람인지도 숫자로 나타내주어야 해서 List에 담은 후, 배열 정렬하고 출력 해준다.

* 새로운 for문을 통해 배열 출력을 하였는데 생각보다 편한 것 같다.