[C++] 백준 16964번 - DFS 스페셜 저지

전형적인 DFS문제이다. 

그런데 이제 Sorting을 곁들인.....

순서대로 DFS를 수행해주면 되지만 인접리스트로 구현한 그래프를 정렬을 하고 DFS를 수행해줘야 한다.

문제해결기법 13주차 A번 문제와 매우 유사한 문제였던 것 같다. 

 

#include <bits/stdc++.h>
#define endl '\n'
#define pii pair<int, int>
using namespace std;

int t;
vector<int> v[100001];
bool visited[100001];
int order[100001];
vector<int> dfs_order;

bool cmp(int& a, int& b) {
	return order[a] < order[b];
}

void dfs(int start) {
	if (visited[start]) return;
	visited[start] = true;
	dfs_order.push_back(start);
	for (auto i : v[start]) if (!visited[i]) dfs(i);
}

void solve() {

	int n; cin >> n;
	for (int i = 0; i < n - 1; i++) {
		int a, b; cin >> a >> b;
		v[a].push_back(b);
		v[b].push_back(a);
	}
	vector<int> temp(n + 1);
	for (int i = 1; i <= n; i++) {
		cin >> temp[i];
		order[temp[i]] = i;
	}
	for (int i = 1; i <= n; i++) {
		sort(v[i].begin(), v[i].end(), cmp);
	}
	dfs_order.push_back(0);
	if (temp[1] == 1) dfs(1);
	if (dfs_order == temp) cout << 1 << endl;
	else cout << 0 << endl;
}


int main() {

	ios::sync_with_stdio(false);
	cin.tie(0); cout.tie(0);

	t = 1; while (t--) solve();

	return 0;
}