[C++] 백준 1764번 - 듣보잡(unordered_map)

문제 링크 : https://www.acmicpc.net/problem/1764

1764번: 듣보잡

 

문제

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

입력

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

 

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

출력

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

 

 

---

 

접근 방법

입력되는 명단에 중복되는 이름이 없으므로 이름을 key로 하는 unordered_map을 사용해서 빠른 탐색을 하도록 하자.

먼저 N명의 듣도 못한 사람 명단을 map에 입력받고,

그 다음 M명의 보도 못한 사람의 명단을 입력받는데 이때, 그 보도 못한 사람의 이름이 듣도 못한 사람 명단에 있는지 검사해서 있으면 ansVec에 push한다.

명단을 사전순으로 출력하기 위해 마지막에 ansVec를 sort 해주면 된다.

 

//
//  Map_BOJ1764_듣보잡.cpp
//  Coding_Test_Practice
//
//  Created by 김난영 on 2021/07/04.
//  Copyright © 2021 KimNanyoung. All rights reserved.
//

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

int main(){
    
    int N, M; cin >> N >> M;
    string str1 = "";
    unordered_map<string, int> umap;
    vector<string> ansVec;
    for(int i = 0; i<N; i++){
        cin >> str1;
        umap[str1] = 0;
    }
    string str2 = "";
    for(int i = 0; i<M; i++){
        cin >> str2;
        if(umap.find(str2)!=umap.end()){    //보도 못한 사람이 듣도 못한 사람 명단에도 있으면
            ansVec.push_back(str2);
        }
    }
    
    sort(ansVec.begin(), ansVec.end());
    cout << ansVec.size() << "\n";
    for(int i = 0; i<ansVec.size(); i++){
        cout << ansVec[i] << "\n";
    }
    
    return 0;
}