[BOJ] #2644. 촌수계산 / 실버2 / 51분 / 성공

Author: Mingguriguri

minjeong/DFSBFS/2024-11-05-[백준]-#2644-촌수계산(BFS).py

@@ -0,0 +1,37 @@
+# 2644-촌수계산. BFS 풀이
+import sys
+from collections import deque
+
+input = sys.stdin.readline
+
+n = int(input())                    # 전체 사람 수
+a, b = map(int, input().split())    # 촌수 계산할 두 사람의 번호
+m = int(input())                    # 부모-자식 관계의 개수
+graph = [[] for _ in range(n + 1)]  # 그래프 초기화
+visited = [False for _ in range(n + 1)]  # 방문 여부 초기화
+for _ in range(m):
+    x, y = map(int, input().split())
+    graph[x].append(y)
+    graph[y].append(x)
+
+
+def bfs(start, end):
+    queue = deque([(start, 0)]) # (노드, 거리) 형태로 초기화
+    visited[start] = True       # 시작 노드 방문 표시
+
+    while queue:
+        v, dist = queue.popleft()
+
+        # 목표 노드에 도달 시 거리 반환
+        if v == end:
+            return dist
+
+        for neighbor in graph[v]:
+            if not visited[neighbor]:
+                visited[neighbor] = True
+                queue.append((neighbor, dist + 1))  # 거리 1 증가하여 큐에 추가
+
+    return -1  # 연결되지 않았을 경우 -1 반환
+
+
+print(bfs(a, b))

minjeong/DFSBFS/2024-11-05-[백준]-#2644-촌수계산(DFS).py

@@ -0,0 +1,36 @@
+# 2644-촌수계산. DFS(백트래킹) 풀이
+import sys
+
+input = sys.stdin.readline
+
+n = int(input())                    # 전체 사람 수
+a, b = map(int, input().split())    # 촌수 계산해야 하는 두 사람의 번호
+m = int(input())                    # 부모 자식들간의 관계의 개수
+graph = [[] for _ in range(n + 1)]  # 그래프
+visited = [False for _ in range(n + 1)]  # 방문여부
+for _ in range(m):
+    x, y = map(int, input().split())
+    # 그래프 연결
+    graph[x].append(y)
+    graph[y].append(x)
+
+result = -1  # 기본값은 -1로 설정하여 찾지 못한 경우에 대비
+
+def backtracking(v, level):
+    global result
+    # 재귀함수 마치는 조건: a,b 촌수를 계산이 되었을 때
+    if v == b:
+        result = level
+        return
+
+    # 탐색
+    for i in graph[v]:
+        if not visited[i]:
+            visited[i] = True
+            backtracking(i, level + 1)  # 자식 노드 방문 level + 1을 넘겨줌으로써 촌수 계산
+            visited[i] = False          # 방문했다면 부모노드 다시 방문 기록 지움
+
+
+visited[a] = True # 초기 노드 방문 체크
+backtracking(a, 0)
+print(result)