[백준/BOJ] 2644번 촌수계산

https://www.acmicpc.net/problem/2644

2644번: 촌수계산

---

 

BFS로 풀었고 오랜만에 BFS로 풀어서 그런지 코드를 구현하는데 좀 오래 걸렸다

앞으로 알고리즘 문제를 자주 풀어야겠다!!!!

 

코드

import sys
from collections import deque
input = sys.stdin.readline

def bfs(start, final):
    queue = deque()
    queue.append(start)

    while queue:
        x = queue.popleft()
        for i in graph[x]:
            if visited[i] == 0:
                visited[i] = visited[x]+1
                queue.append(i)
                if i == final:
                    return
            else:
                continue


n = int(input())
x, y = map(int, input().split())
edge = int(input())
graph = [[] for _ in range(n+1)]
visited = [0 for _ in range(n+1)]

for i in range(edge):
    a, b = map(int, input().split())
    graph[a].append(b)
    graph[b].append(a)

bfs(x, y)
if visited[y] == 0:
    print(-1)
else:
    print(visited[y])