[BOJ 2379 // C++] 트리 탐색하기

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

 

이번에 볼 문제는 백준 2379번 문제인 트리 탐색하기이다.
문제는 아래 링크를 확인하자.

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

2379번: 트리 탐색하기

한국어 번역 지문으로 알기 어려워 추가하면, 이 문제는 단순히 두 트리가 동형인지를 확인하는 문제가 아닌 출발점을 root로 하는 두 rooted tree가 동형인지를 확인하는 문제이다.

 

입력제한이 충분히 작으므로 Tree Isomorphism에 대한 AHU algorithm(Aho-Hopcroft-Ullman algorithm)에 사용되는, 루트가 고정된 트리에 대하여 탐색순서에 무관하게 얻을 수 있는 트리 고유의 문자열을 찾는 코드를 작성해 문제를 해결하자.

 

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

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
using namespace std;

int T;

void solve() {
	string s1, s2; cin >> s1 >> s2;
	vector<string> vec[3000];
	int lv = 0;
	for (auto& l : s1) {
		if (l == '0') lv++;
		else {
			string tmp = "0";
			sort(vec[lv].begin(), vec[lv].end());
			for (auto& s : vec[lv]) tmp += s;
			tmp += "1";
			vec[lv].clear();
			lv--;
			vec[lv].emplace_back(tmp);
		}
	}
	s1 = "";
	sort(vec[0].begin(), vec[0].end());
	for (auto&s:vec[0]) s1 += s;
	vec[0].clear();

	for (auto& l : s2) {
		if (l == '0') lv++;
		else {
			string tmp = "0";
			sort(vec[lv].begin(), vec[lv].end());
			for (auto& s : vec[lv]) tmp += s;
			tmp += "1";
			vec[lv].clear();
			lv--;
			vec[lv].emplace_back(tmp);
		}
	}
	s2 = "";
	sort(vec[0].begin(), vec[0].end());
	for (auto& s : vec[0]) s2 += s;
	vec[0].clear();

	if (s1 == s2) cout << 1 << '\n';
	else cout << 0 << '\n';
}

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

	cin >> T;
	while (T--) solve();
}