[백준 1764] 듣보잡 자바(Java) 풀이

문제 링크 : https://www.acmicpc.net/problem/1764

1764번: 듣보잡

 

[ 문제 ]

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

 

[ 입력 ]

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

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

 

[ 출력 ]

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

 

[ 예제 입력 ]

3 4
ohhenrie
charlie
baesangwook
obama
baesangwook
ohhenrie
clinton

[ 예제 출력 ]

2
baesangwook
ohhenrie

 

[ 풀이 ]

듣도 못한 사람 :

ohhenrie
charlie
baesangwook

 

보도 못한 사람 : 

obama
baesangwook
ohhenrie
clinton

 

처음에 ArrayList로 풀면 된다고 생각했는데 시간초과가 나버렸습니다.. 그래서 HashSet을 사용하였는데 검색해 보니 ArrayList의 contains는 O(N)이 걸리고 HashSet의 contains O(1)이 걸린다고 하네요.

다른 방법으론 map의 getOrDefault를 사용하셔도 잘 풀리실 겁니다!

 

[ 결과 ]

 

[ 정답 코드 ]

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

public class Main {
   static int N;
   static int M;
   static HashSet<String> first;
   static List<String> result;


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

      N = Integer.parseInt(st.nextToken());
      M = Integer.parseInt(st.nextToken());

      first = new HashSet<>();
      result = new ArrayList<>();


      for (int i = 0; i < N; i++) {
         first.add(br.readLine());
      }

      for (int i = 0; i < M; i++) {
         String second = br.readLine();
         if(first.contains(second)){
            result.add(second);
         }

      }

      System.out.println(result.size());
      Collections.sort(result);

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

   }

}