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

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

1764번: 듣보잡

 

문제

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

입력

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

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

출력

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

 

풀이

자료구조 Set(집합) 을 이용하면 비교적 쉬운 풀이가 가능하다

13%에서 틀렸습니다 가 뜬다면 출력조건 (사전순으로 출력)을 지켰는지 확인해보자

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

using namespace std;

int main() {
    set<string> s;
    vector<string> v;
    
    int n, m;
    cin >> n >> m;
    
    string input;
    while(n--) {
        cin >> input;
        s.insert(input);
    }
    while(m--) {
        cin >> input;
        if(s.find(input) != s.end())
            v.push_back(input);
    }
    
    sort(v.begin(), v.end());
    cout << v.size() << "\n";
    for(string i : v)
        cout << i << "\n";
}