[백준/phthon3] 9372번: 상근이의 여행

יוֹסֵף 2023. 11. 6. 08:39

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

9372번: 상근이의 여행

첫 번째 줄에는 테스트 케이스의 수 T(T ≤ 100)가 주어지고, 각 테스트 케이스마다 다음과 같은 정보가 주어진다. 첫 번째 줄에는 국가의 수 N(2 ≤ N ≤ 1 000)과 비행기의 종류 M(1 ≤ M ≤ 10 000) 가

www.acmicpc.net

문제

코드

import sys
from collections import deque
def bfs(x):
    queue = deque([x])
    visited[x] = 1
    cnt = 0
    while queue:
        queue.popleft()
        for i in range(1, n + 1):
            if visited[i] == 0:
                queue.append(i)
                visited[i] = 1
                cnt += 1
    return cnt
t = int(sys.stdin.readline())
for _ in range(t):
    n, m = map(int, sys.stdin.readline().split())
    graph = [[0] * (n + 1) for i in range(n + 1)]
    for i in range(m):
        a, b = map(int, sys.stdin.readline().split())
        graph[b][a] = 1
    visited = [0] * (n + 1)
    print(bfs(1))

저작자표시 (새창열림)