[BOJ] 13023. ABCDE (C++)

• 코딩 초보

---

13023. ABCDE (골드5)

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을 출력한다.

 

 

 

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

int visit[2001]; 
vector <int> graph[2001]; 
int n, m; 
bool res=false; 

void dfs(int node,int count) {
	if (count == 4) {
		res = true; 
		return; 
	}
	visit[node] = true;
	for (int i = 0; i < graph[node].size(); i++) {
		int next = graph[node][i]; 
		if (visit[next]==0 && res==false) {
			dfs(next,count+1); 
		}
	}
	visit[node] = false; 
}


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

	cin >> n>>m;
	for (int i = 0; i < m; i++) {
		int a, b;
		cin >> a >> b; 
		graph[a].push_back(b);
		graph[b].push_back(a); 
	}

	for (int i = 0; i < n; i++) {
		dfs(i, 0); 
		if (res) break; 
	}

	cout << res; 

	

	return 0; 
}

 

 

트리의 깊이(depth)가 4가 되는 경우가 있는지 없는지를 판별하면 된다. 초반에 int graph[2001][2001];로 했었는데 이렇게 되는 경우 dfs 호출에서 시간초과가 발생하는 듯 했다. 그래서 vector을 사용하여 dfs함수를 바꿨다. 

dfs를 호출할때마다 count를 1씩 증가시켜서 재귀호출한다. 호출할때마다, for문을 돌기 전에 count가 4가 되었는지를 판별하고, 4가 되면 함수를 종료하고 결과를 출력토록 했다.