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

문제

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

입력

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

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

출력

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

문제 분석

듣도 못한 사람의 배열, 보도 못한 사람의 배열을 만들고 이 두 배열에서 겹치는 사람의 수와 이름을 사전순으로 출력하는 문제이다. N과 M이 50만 이하의 자연수이므로 이중 for문을 쓴다면 시간 초과가 날 것이 분명하다고 생각했다. 그래서 생각한 것이 출력 또한 사전순으로 출력하기 때문에, 입력받은 이름들을 저장한 배열들을 사전순으로 정렬하고, 이분탐색을 사용한다면 시간복잡도가 훨씬 줄어들어 시간 내에 실행이 가능하다고 생각했다.

작성 코드

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>

using namespace std;

vector<string> notListen;
vector<string> notSee;
vector<string> notSeenLis;

int main(void) {
	ios_base::sync_with_stdio(0);
	cin.tie(0);

	int n, m, cnt = 0;
	string name;
	cin >> n >> m;
	for (int i = 0; i < n; i++) {
		cin >> name;
		notListen.push_back(name);
	}

	for (int i = 0; i < m; i++) {
		cin >> name;
		notSee.push_back(name);
	}
	sort(notListen.begin(), notListen.end());
	sort(notSee.begin(), notSee.end());

	if (notListen.size() > notSee.size()) {
		for (int i = 0; i < m; i++) {
			if (binary_search(notListen.begin(), notListen.end(), notSee[i])) {
				notSeenLis.push_back(notSee[i]);
				cnt++;
			}
		}
	}
	else {
		for (int i = 0; i < n; i++) {
			if (binary_search(notSee.begin(), notSee.end(), notListen[i])) {
				notSeenLis.push_back(notListen[i]);
				cnt++;
			}
		}
	}
	cout << cnt << "\n";

	for (int i = 0; i < notSeenLis.size(); i++) {
		cout << notSeenLis[i] << "\n";
	}
	return 0;
}

코드는 정답!!
if-else 문으로 듣도 못한 배열의 크기와 보도 못한 배열의 크기를 나눠준 이유는 실행 시간을 더 줄여주기 위해 for문을 더 적게 돌도록함에 있어서 이렇게 작성했다.

아쉬운 점

cnt를 사용하지 않고 notSeenLis 배열의 사이즈를 출력해줘도 괜찮았을 것 같다.