[프로그래머스] 네트워크 - python

네트워크

문제

네트워크란 컴퓨터 상호 간에 정보를 교환할 수 있도록 연결된 형태를 의미합니다. 예를 들어, 컴퓨터 A와 컴퓨터 B가 직접적으로 연결되어있고, 컴퓨터 B와 컴퓨터 C가 직접적으로 연결되어 있을 때 컴퓨터 A와 컴퓨터 C도 간접적으로 연결되어 정보를 교환할 수 있습니다. 따라서 컴퓨터 A, B, C는 모두 같은 네트워크 상에 있다고 할 수 있습니다.

컴퓨터의 개수 n, 연결에 대한 정보가 담긴 2차원 배열 computers가 매개변수로 주어질 때, 네트워크의 개수를 return 하도록 solution 함수를 작성하시오.

제한사항

• 컴퓨터의 개수 n은 1 이상 200 이하인 자연수입니다.
• 각 컴퓨터는 0부터 n-1인 정수로 표현합니다.
• i번 컴퓨터와 j번 컴퓨터가 연결되어 있으면 computersi를 1로 표현합니다.
• computer[i] [i]는 항상 1입니다.

 

 

풀이

전형적인 dfs 문제이다.

처음에 인접행렬 그 자체로 푸니까 통과 못하는 테스트 케이스가 있었다. 근데 인접 리스트로 바꿔서 dfs 돌리니까 모든 테스트 케이스를 통과했다. 왜지 ..?

그냥 헷갈리니까 앞으로도 인접 리스트로 바꿔서 편하게 풀어야겠다.

def dfs(start, adj, visited):
    if visited[start]:
        return
 
    visited[start] = True
  
    for next_node in adj[start]:
        if not visited[next_node]:
            dfs(next_node, adj, visited)
​
def solution(n, computers):
    answer = 0
    adj = []
    visited = [False] * n
    
    # 인접 리스트로 변환
    for start in range(n):
        adj.append([i for i, x in enumerate(computers[start]) if x and i != start])
    
    # 방문하지 않은 노드에서 dfs
    for start in range(n):
        if not visited[start]:
            dfs(start, adj, visited)
            answer += 1
    return answer