[백준/C++] 1764번 : 듣보잡

문제

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

입력

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

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

출력

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

풀이

문제 제목 보고 어이없어서 10분간 정지했던 문제ㅎㅋㅋㅋ

set을 사용하면 쉽게 풀 수 있는 문제이다!

#include <iostream>
#include <set>
#include <vector>
#include <algorithm>
using namespace std;

int main(){
    int n, m, ans=0;
    string input;
    cin >> n >> m;

    // 01. 듣도 못한 사람을 set에 등록
    set<string> name; // map보다 빨라짐!(20ms)
    while (n--){
        cin >> input;
        name.insert(input);
    }

    // 02. 보도 못한 사람과 듣도 못한 사람 비교
    // vector에 듣보잡 이름 저장
    vector<string> dupName;
    while (m--){
        cin >> input;
        if (name.count(input)){
            ans++;
            dupName.push_back(input);
        }
    }
    // 03. 듣보잡 이름 사전 순으로 정렬
    sort(dupName.begin(), dupName.end());

    // 출력
    cout << ans << "\n";
    for (int i=0; i<dupName.size();i++){
        cout << dupName[i] << "\n";
    }
}