1. 문제

[Gold V] ABCDE - 13023

문제 링크

성능 요약

메모리: 243712 KB, 시간: 612 ms

분류

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

문제 설명

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

2. 풀이

import sys
sys.setrecursionlimit(100000)

N, M = map(int, input().split())

graph = [[] for _ in range(N + 1)]
visit = [0 for _ in range(N + 100)]

for _ in range(M):
    a, b = map(int, input().split())
    graph[a].append(b)
    graph[b].append(a)

def dfs(i, depth):
    global tf
    visit[i] = 1

    if depth == 4:
        tf = True
        return
    
    for j in graph[i]:
        if visit[j] == 0:
            visit[j] = 1
            dfs(j, depth + 1)
            visit[j] = 0

answer = 0
for i in range(N):
    tf = False
    dfs(i, 0)
    visit[i] = 0

    if tf == True:
        answer = 1
        break

print(answer)