[백준] 13023번 ABCDE - Python / 알고리즘 기초 2/2 - 그래프 1

---

---

---

💡 문제

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

---

💭 접근

처음 문제를 접했을 때 계속 시간초과가 발생하길래 찾아보니 친구가 n명이 되는 경우가 아닌 5명이라도 되는 경우가 있다면 종료를 하는 것이었다..

따라서 기본적인 dfs 문제로, 시작지점을 각 정점마다 시작할 수 있도록 한 후, 하나라도 친구가 5명이 되는 경우가 있다면 break 해주면되는 문제이다.

---

📒 코드

import sys
sys.setrecursionlimit(10**6)

def dfs(start, depth):
    global ans

    if depth == 5:
        ans = True
        return
    
    for num in graph[start]:
        if not visited[num]:
            visited[num] = True
            dfs(num, depth + 1)
            visited[num] = False

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

ans = False
for i in range(n):
    visited[i] = True
    dfs(i, 1)
    visited[i] = False
    if ans:
        print(1)
        break
else:
    print(0)

---

💭 후기

문제를 천천히 읽고 완벽히 이해한 뒤 문제를 풀자...

---

🔗 문제 출처

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

---