[Python] 백준 / ABCDE / 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) 같은 친구 관계가 두 번 이상 주어지는 경우는 없다.

5 5
0 1
1 2
2 3
3 0
1 4

출력

문제의 조건에 맞는 A, B, C, D, E가 존재하면 1을 없으면 0을 출력한다.

1

접근 방식

• 총 5명이 한 방향으로 연결되어있는지 확인하기 위해 각 정점에서 DFS로 탐색하여 깊이가 5 이상일 때 바로 1을 출력하고 아니라면 모두 탐색한 후 0을 출력한다

코드

# url : https://www.acmicpc.net/problem/13023
# 난이도 : gold 5

import sys
sys.setrecursionlimit(10**6) # 백준 재귀제한 해제

n,m = map(int,sys.stdin.readline().split())

graph = [[] for _ in range(n)]

visited = [False]* n
for _ in range(m):
    a,b = map(int,sys.stdin.readline().split())

    graph[a].append(b)
    graph[b].append(a)


switch = False # 깊이가 5 이상일 때 켜지는 변수
def dfs(node,depth,visited):
    
    global switch
	
    # 깊이가 5 이상일 때 switch를 키고 리턴한다
    if depth >= 5:
        switch = True
        return
    

    
    for i in graph[node]:
        if visited[i] == False:
            visited[node] = True
            dfs(i,depth+1,visited)
            visited[node] = False

# 모든 정점에 대해 검색한다
for i in range(n):
    visited[i] = True
    dfs(i,1,visited)
    visited[i] = False
	
    # 스위치가 켜져있으면 1을 출력하고 멈춘다
    if switch == True :
        print(1)
        break

# 모든 정점을 검색해도 스위치가 켜져있지 않았다면 0을 출력한다
else:
    print(0)