JAVA 백준 14425 문자열 집합 (집합과 맵)

 

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

 

---

문제

1. 첫째 줄에 집합 S의 크기 N과 주어질 문자열의 개수 M ( 1 <= N <= 10000, 1 <= M <= 10000  ) 이 입력된다.
2. 둘째 줄부터 N개의 줄에 집합 S에 포함될 문자열이 N개 입력된다.
3. 그 다음 줄부터 M개의 줄에 집합 S에 포함된 문자열과 비교할 문자열 M개가 입력된다.
4. 집합 S와 M개의 문자열을 하나씩 비교하며 집합 S에 포함되어 있는 지 개수를 세어 출력한다.

---

 
 
정답 코드

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

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());

        HashSet<String> set = new HashSet<>();

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

        int result = 0;

        for (int i = 0; i < M; i++) {
            if (set.contains(br.readLine())) {
                result++;
            }
        }

        System.out.println(result);
    }
}

 
중복은 없지만 포함되었는지 비교하는 건 Hash가 빠르므로 Hash 관련 자료구조 중에 필요로 하는 것이 한 가지인 HashSet을 사용하기로 했다.
 
집합 S의 문자열을 HashSet에 모두 넣고 M개의 문자열을 받으며, HashSet에 있다면 출력할 변수를 1 증가시키기만 하면 된다.