[백준] 문자열 처리 - 1764. 듣보잡

문제

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

 

입력

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

 

출력

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

 

예제 입력1

3 4
ohhenrie
charlie
baesangwook
obama
baesangwook
ohhenrie
clinton

 

예제 출력1

2
baesangwook
ohhenrie

 

생각

처음에 find() 함수를 사용해보는 것으로 했다. O(N)의 시간복잡도가 나와 시간 초과가 나온다. 

binary_search()를 이용해서 풀어야한다. (시간복잡도 생각해야되는 문제 => dp, 분할정복! )

push_back을 사용하는 것보다 resize() 를 사용해서 미리 메모리를 할당하면 메모리 사용을 줄일 수 있다.

백준에서 실행해보니 200KB나 차이가 났다.

 

작성한 코드

#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
 
using namespace std;
 
int main(int argc, const char *argv[])
{
    // insert code here...
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
 
    int n, m;
    vector<string> a;      //듣도 못한 사람
    vector<string> result; //듣도 보도 못한 사람
 
    cin >> n >> m;
    a.resize(n);
 
    for (int i = 0; i < n; i++)
    {
        cin >> a[i];
    }
 
    // 이진 탐색을 위한 정렬
    sort(a.begin(), a.end());
 
    for (int i = 0; i < m; i++)
    {
        string s;
        cin >> s;
        if (binary_search(a.begin(), a.end(), s))
        {
            result.push_back(s);
        }
    }
 
    sort(result.begin(), result.end());
 
    cout << result.size() <<"\n";
    for (int i = 0; i < result.size(); i++)
    {
        cout << result[i] << "\n";
    }
 
    return 0;
}