[BOJ 25758 // C++] 유전자 조합

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

 

이번에 볼 문제는 백준 25758번 문제인 유전자 조합이다.
문제는 아래 링크를 확인하자.

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

25758번: 유전자 조합

알파벳 대문자는 총 26가지이므로, 생물의 두 형질중 첫 번째 유전자로 'A'에서 'Z'까지 각 하나를 고정했을 때 다른 형질로 골라질 수 있는 경우의 수를 전부 탐색하는 것으로도 문제를 충분히 해결할 수 있다. 위와 같은 탐색의 시간복잡도는 O((알파벳의 가짓수) * (문자열 개수))가 된다는 점을 확인하자.

 

아래는 제출한 소스코드이다.

#include <iostream>
#include <string>
#include <set>
using namespace std;

int N;
string arr[100000];
set<char> ans;

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

	cin >> N;
	for (int i = 0; i < N; i++) cin >> arr[i];
	for (char x = 'A'; x <= 'Z'; x++) {
		int cnt = 0;
		for (int i = 0; i < N; i++) {
			if (arr[i][0] == x) cnt++;
		}

		if (cnt > 1) {
			for (int i = 0; i < N; i++) {
				ans.insert(max(x, arr[i][1]));
			}
		}
		else if (cnt == 1) {
			for (int i = 0; i < N; i++) {
				if (arr[i][0] == x) continue;
				ans.insert(max(x, arr[i][1]));
			}
		}
	}

	cout << ans.size() << '\n';
	for (auto& x : ans) cout << x << ' ';
}