[백준] ABCED c++

2024. 10. 23. 13:23코딩테스트

ABCDE 

 
시간 제한메모리 제한제출정답맞힌 사람정답 비율
2 초 512 MB 58785 18708 12413 28.890%

문제

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


DFS를 이용해 깊이가 5가 될때까지 탐색이 된다면 문제에 맞는 관계가 있는것이다.

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

static vector<vector<int>> A;
static vector<bool> Visited;
static bool arrive;
void DFS(int now, int deep);

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

	int N, M;

	cin >> N >> M;

	A.resize(N);
	Visited = vector<bool>(N, false);

	for (int i = 0; i < M; i++)
	{
		int s, e;
		cin >> s >> e;
		A[s].push_back(e);
		A[e].push_back(s);
	}

	for (int i = 0; i < N; i++)
	{
		DFS(i, 1);
		if (arrive)
		{
			break;
		}
	}

	if (arrive)
	{
		cout << "1";
	}
	else
	{
		cout << "0";
	}
}

void DFS(int now, int deep)
{
	if (deep == 5)
	{
		arrive = true;
		return;
	}

	Visited[now] = true;

	for (int i : A[now])
	{
		if (!Visited[i])
		{
			DFS(i, deep + 1);
		}
	}

	Visited[now] = false;
}

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

 

'코딩테스트' 카테고리의 다른 글

[백준] 미로 탐색하기 c++  (0) 2024.10.23
[백준] DFS와 BFS c++  (0) 2024.10.23
[프로그래머스] N개의 최소공배수  (0) 2024.10.21
[프로그래머스] 점프와 순간이동  (0) 2024.10.21
[백준] 신기한 소수 c++  (1) 2024.10.19