본문 바로가기

백준/C++

[Baekjoon/C++] 1764번 - 듣보잡

[백준] Baekjoon Online Judge

문제로 이동

 

문제

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

 

입력

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

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

 

출력

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

 


예제 입력 예제 출력
3 4
ohhenrie
charlie
baesangwook
obama
baesangwook
ohhenrie
clinton
2
baesangwook
ohhenrie

풀이

#include <iostream>
#include <algorithm>
#include <map>
using namespace std;

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	int N, M; // 듣도 못한 사람의 수 N, 보도 못한 사람의 수 M
	cin >> N >> M;

	map <string, int> noHear; // 듣도 못한 사람 명단
	string* noHearSee = new string[N]; // 듣도 보도 못한 사람 명단
	int idx = 0; // 듣보잡의 수

	for (int i = 0; i < N; i++) {
		string s;
		cin >> s;

		noHear.insert({ s, i });
	}

	// 듣도 보도 못한 사람의 명단을 구하기
	for (int i = 0; i < M; i++) {
		string s;
		cin >> s;

		if (noHear.find(s) != noHear.end())
			noHearSee[idx++] = s;
	}

	sort(noHearSee, noHearSee + idx);

	// 듣보잡의 수와 그 명단을 사전순으로 출력
	cout << idx << endl;
	for (int i = 0; i < idx; i++)
		cout << noHearSee[i] << "\n";

	return 0;
}

map을 사용해서 문제를 풀었다. 우선 map에 듣도 못한 사람의 명단을 저장했다. 그 뒤 보도 못한 사람의 명단을 하나씩 입력 받았는데, 이때 입력 받은 사람의 이름이 map에 존재한다면 noHearSee[]에 저장했다. 이 과정이 끝나면 noHearSee[]를 정렬하고, 답을 출력했다.