[백준 2606-Graph] 바이러스

CHOI YUN HO·2021년 5월 23일
0

알고리즘 문제풀이

목록 보기
35/63

📃 문제 설명

바이러스

[문제 출처 : 백준]

👨‍💻 해결 방법

1번 컴퓨터랑 연결되어 있는 것들의 수를 구하면 되는 간단한 문제였다

DFS, BFS 둘 다 가능하겠지만 난 DFS가 좋앙~

문제에서 주어진 대로 그래프를 만들고,
DFS함수 구현해서 1부터 탐색하면 끝

재밌당 히히

👨‍💻 소스 코드

N = int(input())
linkNum = int(input())

graph = [[0] * (N + 1) for _ in range(N + 1)]
visited = []

for _ in range(linkNum):
    a, b = input().split()
    a, b = int(a), int(b)
    graph[a][b], graph[b][a] = 1, 1

def dfs(v):
    visited.append(v)
    for i in range(1, N + 1):
        if i not in visited and graph[v][i] == 1:
            dfs(i)

dfs(1)
print(len(visited) - 1)



profile
가재같은 사람

0개의 댓글