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

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

1764번: 듣보잡

문제

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

입력

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

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

출력

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

소스코드

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

using namespace std;

int main() {
    int N,M;
    set<string> s; //set() 선언
    string people;
    int cnt = 0;

    cin >> N >> M; 

    string *arr = new string[N+M];

    for(int i = 0; i < N; i++){
        cin >> people;
        s.insert(people); //듣도 못한 사람들 set()함수에 삽입
    }

    for(int i = 0; i < M; i++){
        cin >> people;
        if(s.find(people) != s.end()){ // 듣도못한 사람들 중 겹치는 사람 찾기
            arr[cnt] = people;
            cnt++; // 겹치는 사람 수 카운팅
        }
    }

    sort(arr,arr+cnt); //사전순 정렬

    cout << cnt << "\n";
    for(int i = 0; i < cnt; i++){
        cout << arr[i] << "\n";
    } 
    return 0;
}

풀이 과정

set()함수를 이용해 s 선언후 반복문을 이용해 듣도 못한 사람들을 s에 삽입한다.

이 중 보도못한 사람들과 중복되는 인원을 찾기 위해 배열 arr을 동적할당으로 선언한다.

s.find()함수를 이용해 겹치는 사람들을 찾고 이들을 arr배열에 저장한다. 또한 이들의 숫자를 세기위해 겹칠때마다 cnt에 +1을 한다.

사전 순으로 출력하라 했으므로 sort()함수를 통하여 정렬한다.

이후 cnt와 함께 정렬된 arr배열을 출력한다.