[프로그래머스] 네트워크 (Python)

link 🔗 https://programmers.co.kr/learn/courses/30/lessons/43162

코딩테스트 연습 - 네트워크

1️⃣ check 배열을 활용하여 값 구하기

def solution(n, computers):
    cnt = 1
    check = [0]*n
    def dfs(now):
        nonlocal cnt
        for j in range(n):
            if computers[now][j] == 1 and check[j] == 0:
                check[j] = cnt
                dfs(j)
    
    for idx in range(n):
        if check[idx] == 0:
            check[idx] = cnt
            dfs(idx)
            cnt += 1
    
    answer = max(check)
    
    return answer