[Gold / 13023] ABCDE

문제 링크

---

🔷 분류

그래프 이론, 그래프 탐색, 깊이 우선 탐색, 백트래킹

✒️ 문제 설명

BOJ 알고리즘 캠프에는 총 N명이 참가하고 있다. 사람들은 0번부터 N-1번으로 번호가 매겨져 있고, 일부 사람들은 친구이다.

오늘은 다음과 같은 친구 관계를 가진 사람 A, B, C, D, E가 존재하는지 구해보려고 한다.

• A는 B와 친구다.
• B는 C와 친구다.
• C는 D와 친구다.
• D는 E와 친구다.

위와 같은 친구 관계가 존재하는지 안하는지 구하는 프로그램을 작성하시오.

⬅️ 입력

첫째 줄에 사람의 수 N (5 ≤ N ≤ 2000)과 친구 관계의 수 M (1 ≤ M ≤ 2000)이 주어진다.

둘째 줄부터 M개의 줄에는 정수 a와 b가 주어지며, a와 b가 친구라는 뜻이다. (0 ≤ a, b ≤ N-1, a ≠ b) 같은 친구 관계가 두 번 이상 주어지는 경우는 없다.

➡️ 출력

문제의 조건에 맞는 A, B, C, D, E가 존재하면 1을 없으면 0을 출력한다.

💻 코드 (C++)

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

vector<vector<int>> A(2000);
vector<bool> visited(2000, false);
bool arrive = false;

void DFS(int node, int depth) {
	if(depth == 5 || arrive) {
		arrive = true;
		return;
	}
	visited[node] = true;
	for(int i : A[node]) {
		if(!visited[i]) {
			DFS(i, depth + 1);
		}
	}
    visited[node] = false;
}

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	int n, m;
	cin >> n >> m;
	for(int i = 0; i < m; i++) {
		int a, b;
		cin >> a >> b;
		A[a].push_back(b);
		A[b].push_back(a);
	}
	for(int i = 0; i < n; i++) {
		DFS(i, 1);
		if(arrive) break;
	}
	if(arrive) cout << 1;
	else cout << 0;
	return 0;
}