BOJ 1764 듣보잡

By kuper0201

Published in 알고리즘

2023-05-25

1 min read

문제

입력

풀이

문제

BOJ 1764 듣보잡 바로 가기

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

입력

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

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

풀이

단순한 구현 문제로 듣도 못한 사람의 명단과 보도 못한 사람의 명단 중 겹치는 사람의 명단을 출력하는 문제이다.

생각할 수 있는 가장 간단한 풀이는 배열을 이용해 명단을 저장 한 뒤, 순차 탐색을 하는 방법이다.

하지만 이러한 풀이는 $\mathcal{O}(N^{2})$ 의 시간 복잡도를 가지므로 N의 제한(500,000 이하의 자연수)으로 인해 시간 초과가 발생한다.

Set을 이용하면 $\mathcal{O}(1)$ 의 시간 복잡도만으로 탐색이 가능하다.

따라서 해당 문제는 Set을 이용해 듣도 못한 사람의 명단을 저장하고, 명단에 보도 못한 사람이 존재하는지 판별하면 해결할 수 있다.

코드 보기(Java)

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        HashSet<String> list = new HashSet<>(); // 듣도 못한 사람의 명단
        ArrayList<String> res = new ArrayList<>(); // 듣보잡 명단 저장할 리스트

        // 명단의 수 입력
        String[] str = br.readLine().split(" ");

        // 듣도 못한 사람 명단 추가
        int M = Integer.parseInt(str[0]);
        for(int i = 0; i < M; i++) {
            list.add(br.readLine());
        }

        // 보도 못한 사람 입력
        int N = Integer.parseInt(str[1]);
        for(int i = 0; i < N; i++) {
            String tmp = br.readLine();
            // 듣도 못한 사람 명단에 보도 못한 사람이 존재하는 경우 => 듣보잡
            if(list.contains(tmp)) res.add(tmp);
        }

        // 사전순 출력을 위한 정렬
        Collections.sort(res);

        // 인원수와 명단 출력
        System.out.println(res.size());
        for(String out : res) System.out.println(out);
    }
}

코드 보기(C++)

#include <iostream>
#include <algorithm>
#include <vector>
#include <set>

#define fastio ios_base::sync_with_stdio(false); cin.tie(nullptr); cout.tie(nullptr)
#define endl '\n'

using namespace std;

int main() {
    fastio;
    
    set<string> list; // 듣도 못한 사람의 명단
    vector<string> res; // 듣보잡 명단 저장할 리스트

    // 명단의 수 입력
    int M, N;
    string s;
    cin >> M >> N;
    
    // 듣도 못한 사람 명단 추가
    for(int i = 0; i < M; i++) {
        cin >> s;
        list.insert(s);
    }

    // 보도 못한 사람 입력
    for(int i = 0; i < N; i++) {
        cin >> s;
        // 듣도 못한 사람 명단에 보도 못한 사람이 존재하는 경우 => 듣보잡
        if(list.find(s) != list.end()) res.push_back(s);
    }

    // 사전순 출력을 위한 정렬
    sort(res.begin(), res.end());

    // 인원수와 명단 출력
    cout << res.size() << endl;
    for(string out : res) cout << out << endl;
    
    return 0;
}

코드 보기(Kotlin)

import java.util.*
import java.io.*

fun main(args: Array<String>) = with(System.`in`.bufferedReader()) {
    var list: MutableSet<String> = mutableSetOf<String>() // 듣도 못한 사람의 명단
    var res = ArrayList<String>() // 듣보잡 명단 저장할 리스트
    
    // 명단의 수 입력
    var str = readLine().split(" ")

    // 듣도 못한 사람 명단 추가
    var M = str[0].toInt()
    for (i in 0 until M) {
        list.add(readLine())
    }

    // 보도 못한 사람 입력
    var N = str[1].toInt()
    for (i in 0 until N) {
        var tmp = readLine()
        // 듣도 못한 사람 명단에 보도 못한 사람이 존재하는 경우 => 듣보잡
        if(list.contains(tmp)) res.add(tmp)
    }

    // 사전순 출력을 위한 정렬
    res.sort()
    
    // 인원수와 명단 출력
    println(res.size)
    for (out in res) println(out)
}

코드 보기(Python)

from sys import stdin

def main():
    list = set() # 듣도 못한 사람의 명단
    res = []  # 듣보잡 명단 저장할 리스트

    # 명단의 수 입력
    str = stdin.readline().split(' ')

    # 듣도 못한 사람 명단 추가
    M = int(str[0])
    for i in range(M):
        list.add(stdin.readline().strip())
        
    # 보도 못한 사람 입력
    N = int(str[1])
    for i in range(N):
        tmp = stdin.readline().strip()
        # 듣도 못한 사람 명단에 보도 못한 사람이 존재하는 경우 = > 듣보잡
        if tmp in list:
            res.append(tmp)

    # 사전순 출력을 위한 정렬
    res.sort()

    # 인원수와 명단 출력
    print(len(res))
    for out in res:
        print(out)

if __name__ == "__main__":
    main()

코드 보기(Swift)

import Foundation

func main() {
    var list = Set<String>() // 듣도 못한 사람의 명단
    var res = Array<String>() // 듣보잡 명단 저장할 리스트
    
    // 명단의 수 입력
    var str = readLine()!.split(separator: " ")
    
    // 듣도 못한 사람 명단 추가
    var M = Int(str[0])!
    for i in 0..<M {
        list.insert(readLine()!)
    }
    
    // 보도 못한 사람 입력
    var N = Int(str[1])!
    for i in 0..<N {
        var tmp = readLine()!
        // 듣도 못한 사람 명단에 보도 못한 사람이 존재하는 경우 => 듣보잡
        if list.contains(tmp) {
            res.append(tmp)
        }
    }
    
    // 사전순 출력을 위한 정렬
    res.sort()
    
    // 인원수와 명단 출력
    print(res.count)
    for out in res {
        print(out)
    }
}

main()