[백준] 13023번 - ABCDE (C++)

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

 

 

풀이과정

• vector node : 실질적으로 노드간의 간선정보가 담기는 부분
• bool visited : 방문한 노드인지 아닌지 확인하는 부분
• flag : 5명이 일자로 연결된 경우를 확인해주는 파트

 

dfs와 cktracking 을 사용하여 노드를 이미 방문했는데 돌아가야하는 경우 visitied를 false로 바꿔주는 방식으로 해결.

 

code

#include <iostream>
#include <vector>
#include <stack>

std::vector <int> node[2001];
bool visited[2001];
bool flag;

void	dfs(int cnt, int n, int count)
{
	if (flag) return ;
	if (count == 5 && !flag)
		flag = true;
	visited[cnt] = true;
	for (int i = 0; i < node[cnt].size(); i++) {
		int next = node[cnt][i];
		if (!visited[next]) {
			dfs(next, n, count + 1);
		}
	}
	visited[cnt] = false;
}

int		main(void)
{
	int n, m, a, b;

	std::cin >> n >> m;
	for (int i = 0; i < m; i++) {
		std::cin >> a >> b;
		node[a].push_back(b);
		node[b].push_back(a);
	}
	for (int i = 0; i < n; i++) {
		dfs(i, n, 1);
	}
	if (flag) std::cout << "1" << "\n";
	else std::cout << "0" << "\n";

	return (0);
}

 

후기

상당히 어렵다. 나중에 다시 도전해봐야할 문제