[Algorithm] BaekJoon : 13023. ABCDE by Python

엄희관·2021년 3월 3일
0

Algorithm

목록 보기
110/128
post-thumbnail

[문제 바로가기] https://www.acmicpc.net/problem/13023

📌문제 설명

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


💡 문제 풀이

문제를 너무 간략하게 적어서 출제자의 의도를 이해하지 못했다...

처음에는 A - B - C - D - E ... N-1까지 모두 친구관계임을 찾는줄 알고 서로소 집합을 만들었었다. (당연히... "실패")

하지만, 문제에서 원하는 요구사항은 A → B → C → D → E의 관계가 있는 즉, 깊이 5의 친구관계 유무를 파악하는 것이었다.

5인 "깊이"를 찾아야하므로 DFS 알고리즘으로 해결하였다.

0번부터 N-1번까지 DFS로 탐색하면서 깊이 5만큼의 친구관계가 존재하면 1을 출력!
모든 탐색을 마칠때까지 관계가 존재하지 않았다면 0을 출력하면 된다.

코드는 다음과 같다.

import sys

def dfs(node, depth, visited):
    global flag
    if depth == 5: # 깊이 5의 친구 관계가 존재하면 True
        flag = True
        return
    for idx in relationship[node]:
        if not visited[idx]:
            visited[idx] = True
            dfs(idx, depth+1, visited)
            visited[idx] = False

N, M = map(int, input().split())
relationship = {i:[] for i in range(N)}
for _ in range(M):
    s, e = map(int, sys.stdin.readline().split())
    relationship[s].append(e)
    relationship[e].append(s)

flag = False # 조건에 맞는 '친구 관계' 존재 유무
for node in range(N):
    if not flag:
        visited = [False] * N
        visited[node] = True
        dfs(node, 1, visited)
    else: # 조건에 맞는 친구 관계를 찾았으면 탐색을 멈춘다.
        break
print(1) if flag else print(0)

알고리즘은 역시 문제를 제대로 이해하는게 가장 중요하고 어렵다...😂

profile
허브

0개의 댓글