[프로그래머스] (Python) 네트워크 (lv.3) (DFS, 재귀)

파이썬... 입니다... 썸네일은 거짓말이에요 ㅠㅠ 스위프트 아닙니다.
SK플래닛 시험보기 직전.. 약 6시간전에 Swift언어 응시가 불가하다는 통보를 받았슴다. 하하 java / js / python만 보더군요.. 이번 채용에서는 저 세 언어를 다루는 직무에 TO가 낫나봐요...? 네이티브를 안쓰는건가...
그래서 부랴부랴 파이썬으로 간단하게나마 시험준비겸 문제를 풀어봤슴다. 하하 양해 부탂쓰

⚫️ 문제

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

프로그래머스

 

⚫️ 나의 풀이

문제를 보자마자 DFS임을 파악했다. 하나의 노드로 들어가서, 가장 깊은곳까지탐색하고 그 연결된 것들의 count를 세어주면 된다고 파악했기 때문이다! DFS문제의 정석이라고 느껴졌다. 쉬우니까 바로 구현해버리기

근데,,, dfs 돌릴때 새로 갱신되는 computerNum을 넘겨주어야하는데 초기값을 계속 넘겨줘서 재귀 제한에 걸려서 짜증나서 미쳐버리늕루 알았다 ㅋㅋㅋㅋㅋ 결국 컴퓨터는 잘못이없고 내문제엿다^^^ 파이썬 욕할뻔...

⚫️ 정답코드

def solution(n, computers):

    visited = [False] * (n)
    answer = 0

    # dfs 함수
    def dfs(computerNumber):
        visited[computerNumber] = True

        for i in range(n):
            if i != computerNumber and computers[computerNumber][i] == 1:
                if visited[i] == False:
                    dfs(i)

    # 컴퓨터 하나씩 start 지점으로 다 방문해야함
    for computerNumber in range(n):
        if visited[computerNumber] == False:
            dfs(computerNumber)
            answer += 1
    return answer