< DFS > 13023번 ABCDE with 파이썬

 

문제

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

정답비율

28.949%

알고리즘 분류

 #그래프 #그래프이론 #그래프탐색 #깊이우선탐색 #DFS

 

import sys
input = sys.stdin.readline

def dfs(num, depth):
    if depth >= 4 :
        return 4
    for new in graph[num]:
        if visited[new] :
            continue
        visited[new] = True
        tmp = dfs(new, depth + 1)
        visited[new] = False
        if tmp and tmp >= 4:
            return 4

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

for i in range(n):
    visited = [False] * n
    d = 0
    if not graph[i] :
        continue
    visited[i] = True
    d = dfs(i, d)
    if d :
        print(1)
        quit()
print(0)

 

 - 문제에서 '5명'이 친구가 되는지 구해보려고 한다고 적혀있었으면 좋았을 것 같다. 처음에는 문제가 무슨 의미인지 헷갈렸다.

 

 - 어쨌든 깊이가 4 이상이 되면 1을 출력하는 문제이다.

 

 - 처음에는 dfs함수 내에 visited[new] = False라는 초기화 구문 없이 제출했고, 오답판정을 받았다. 하나의 원소에 대해 탐색후 visited를 완전 초기화하면 될 것이라고 생각했고, 예시도 모두 제대로 동작했다. 하지만 다음과 같은 반례를 도출하였다.

 

6 6
0 1
0 2
0 5
3 5
4 5
3 4
정답 : 1

 

 - 위의 입력에서 0이 나왔던 이유는 3에서 5를 먼저 탐색하게되면 3-5-0-1 이 되어 깊이가 모자라다. 그리고 3-4-5-0-1을 탐색해서 정답을 출력해야하지만 5를 이미 탐색했다고 판단하여 3-4 에서 끝이난다.

 

 - 따라서 dfs가 끝을 찍고 올 때마다 그 원소의 visited를 초기화해주어야한다. 

 

---

 

참고

 

13023번: ABCDE

 

---