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

 

📑출력

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

 

📚 문제 해석

 

문제에서 말하는 친구 사이는 5명이 연결되어 있는 경우를 말한다. 

그림 1

그림 1은 0은 친구 0 -> 1 -> 2 -> 3 -> 4 이렇게 연결되어 있다. 때문에 문제 조건에 맞는 친구사이다.

 

그림 2

그림 2도 2 또는 4에서 시작하면 문제 조건에 맞는 친구사이가 될 수 있다.

2 -> 3 -> 0 -> 1 -> 4

4 -> 1 -> 0 -> 3 -> 2

 

그림 3

그림 3은 어디서 시작하여도 친구사이가 3명이고 5명이 될 순 없다.

때문에 문제 조건에 맞는 친구사이는 없다.

 

해당문제는 DFS로 접근하였다. 깊이가 4일때 정답이 1이 된다는 것을 알 수 있다. (깊이가 4라는 것은 노드(친구)는 5개)

 

💻 코드

import sys
sys.setrecursionlimit(10**4)
input = sys.stdin.readline

def dfs(graph, v, visited, cnt):
    global ans
    if cnt == 4: # 깊이가 4라면 ans를 1로 초기화
        ans = 1
        return 
    for i in graph[v]:
        if not visited[i]:
            visited[i] = True
            dfs(graph, i, visited,cnt+1)
            visited[i] = False # 다시 방문해보기 위해 False 로 (백트래킹?)

n,m = map(int,input().rstrip().split())
graph = [[] for i in range(n)]
ans = 0
# 그래프 초기화
for _ in range(m):
    a,b= map(int,input().rstrip().split())
    graph[a].append(b)
    graph[b].append(a)

# 연결 간선이 4 이상일 때만 해당 문제의 조건이 성립할 수 있다.
if m >= 4:
    for node in range(n):
        visited = [False] * n
        visited[node] = True # 시작 노드 방문 처리
        dfs(graph, node, visited, 0) # DFS 시작
        if ans == 1: # 깊이가 4인게 있다면 멈춤
            break
print(ans)

 

🍭 시행착오

인간승리

런타임 에러는 sys.setrecursionlimit을 하지 못해서 그랬다 치고, 첫번째 틀렸습니다의 이유는 바로visited[i] = False를 하지 않았기 때문이다. 

한번 방문 했던 곳은 다시 방문하지 않아야하는 게 맞지만, 다른 경로로도 가봐야 하기 때문에 다시 False로 초기화 해준 것이다. n과 m 시리즈에서 비슷한 경험을 한 적이 있는 것 같다.

 

그렇게 해서 False 를 주었더니 이제 시간초과가 나왔다..ㅎ

다시 멘붕이 왔지만 맘을 다 잡고 pypy로 제출하였지만 그래도 결과는 변함없었다. 그래서 나중에 다시 해볼까하다가 DFS 함수 구현 부분에서 시간을 줄일 수 있는 부분을 발견하였다. 

if cnt == 4 이부분인데, 나의 시간 초과 났던 코드는 이렇게 정답이 1로 판별 나도 계속 끝까지 탐색을 하는 형태였다. 그래서 cnt ==4 라면 ans 초기화 해준뒤 return 하였더니 시간 초과 문제가 해결되었다!!!ㅎㅎ

 

🔗 문제링크 ABCDE