백준 1764 문제 - 듣보잡

문제출처 : https://www.acmicpc.net/problem/1764 

 

문제

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

 

입력

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

 

출력

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

 

테스트케이스

입력	출력
3 4 ohhenrie charlie baesangwook obama baesangwook ohhenrie clinton	2 baesangwook ohhenrie

 

이 문제를 풀기위해서는 셋(Set) 자료구조와 정렬(Sort) 를 알아야 한다.

우선 듣도 못한 집단을 A 라고 하고 보도 못한 집단을 B라고 하면 듣도 & 보도 못한 집단은 A 집단과 B집단의 교집합이 될 것이다.

 

Note

A ∈ { 듣도 못한 집합 명단 } = { ohhenrie, charlie , baesangwook}
B ∈ { 보도 못한 집합 명단 } = { obama, baesangwook, ohhenrie, clinton}

A ∩ B ∈ { 보도 & 듣도 못한 집합 명단 } = { baesangwook, ohhenrie} -> 오름차순 정렬 

 

 

답은 아래와 같고 셋 자료구조와 정렬을 구현하거나 내장된(built-in)된 것을 활용하면 쉽게 풀수 있다.

 

 
import java.util.HashSet;
import java.util.Set;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
 
 
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]); // 보도 못한 사람 수
 
        Set<String> A = new HashSet<>(); // 듣도 못한 집단
        Set<String> B = new HashSet<>(); // 보도 못한 집단
 
        for(int i = 0 ; i < N ; i++) {
            A.add(br.readLine());
        }
 
        for(int i = 0 ; i < M ; i++) {
            B.add(br.readLine());
        }
 
        // 교집합을 구하고
        Set<String> intersectOfAB = intersect(A, B);
 
        // 듣도&보도 못한 명단 사이즈를 출력하고
        System.out.println(intersectOfAB.size());
        // 사전순으로 정렬해서 출력해라.
        intersectOfAB.stream().sorted().forEach(System.out::println);
 
 
    }
 
    // 집합 s1, s2의 교집합을 구한다.
    public static <E>
    Set<E> intersect(Set<E> s1, Set<E> s2) {
        Set<E> s = new HashSet<>(s1);
        s.retainAll(s2);
        return s;
    }
 
}