[BOJ/백준 14425][C++] 문자열 집합

문제

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

 

문제 내용

n개의 문자열로 이루어진 집합에, 그 뒤로 주어지는 문자열들이 몇 개나 포함되어있는지를 구하는 문제이다.

위 사진에서 빨간색으로 묶여있는 문자열들 안에, 그 뒤로 이어지는(파란색) 문자열의 포함 여부를 판단하면 된다.

 

문제 풀이

문제 자체는 간단하다.
다만 n의 입력 범위가 1~10000, m의 입력 범위가 1~10000이라는 점에서 단순하게 탐색하면 안 된다.
m개 입력되는 문자열을 n개의 집합에서 처음부터 매번 찾을 수는 없기 때문이다. (시간 초과 난다)

그렇기 때문에 이 문제에서는 이진 탐색을 사용했다.
binary_search()는 정렬된 상태에서 사용이 가능하기 때문에, n개의 문자열 집합을 입력받고 나서 sort()로 정렬을 한 번 시켜준다.

그 뒤에 m개의 문자열을 입력받으면서, 매번 이진 탐색을 해보면 된다.
그리고 있으면 count값을 증가시키면 된다.

 

코드

#include <iostream>
#include <algorithm> //binary_search() : 정렬된 상태에서 사용
#include <vector>
using namespace std;

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

	//입력
	int n, m;
	cin >> n >> m;
	vector<string>v1(n); //듣도 못한 사람
	vector<string>result;
	for (int i = 0; i < n; i++) cin >> v1[i];

	//문제 해결
	sort(v1.begin(), v1.end());

	string mm;
	int count = 0;
	for (int i = 0; i < m; i++) {
		cin >> mm;
		if (binary_search(v1.begin(), v1.end(), mm)) count++;
	}

	//결과 출력
	cout << count << '\n';
}

 

문제 코드 설명

1) 문제 해결

binary_search()에서 마지막 인자는 내가 찾고자 하는 것을 넣어주면 된다.