나의 개발일지

[프로그래머스] Lv.3 네트워크 (DFS) [Python, 파이썬] 본문

프로그래머스

[프로그래머스] Lv.3 네트워크 (DFS) [Python, 파이썬]

YoonJuHan 2023. 3. 12. 13:26
def dfs(x, computers, visit):
    visit[x] = 1 # 방문 처리
    for a, b in enumerate(computers[x]):
        if b == 1 and not visit[a]:
            dfs(a, computers, visit)

def solution(n, computers):
    visit = [0] * n		# 방문 확인 리스트 생성
    network = 0			# 네트워크의 수

    for i in range(n):	 # 컴퓨터 개수 만큼 반복
        if not visit[i]: # 방문하지 않은 컴퓨터 (visit[i] == 0)
            dfs(i, computers, visit)	# dfs 함수 진입
            network += 1
    return network
Comments