프로그래머스 - 네트워크(level 3)

프로그래머스 - level 3 네트워크

문제 설명

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

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

제한사항

• 컴퓨터의 개수 n은 1 이상 200 이하인 자연수입니다.

• 각 컴퓨터는 0부터 n-1인 정수로 표현합니다.

• i번 컴퓨터와 j번 컴퓨터가 연결되어 있으면 computers[i][j]를 1로 표현합니다.

• computer[i][i]는 항상 1입니다.

입출력 예

 

n	computers	return
3	[[1, 1, 0], [1, 1, 0], [0, 0, 1]]	2
3	[[1, 1, 0], [1, 1, 1], [0, 1, 1]]	1

 

입출력 예 설명

예제 #1
아래와 같이 2개의 네트워크가 있습니다.

예제 #2
아래와 같이 1개의 네트워크가 있습니다.

---

문제풀이

이전에 설명했던 BFS, DFS 관련 문제이다. 심플하게 해석하자면 그래프의 갯수가 몇개가 되는지 구하는 것이다.

이를 구하기 위한 절차를 살펴보면

1. BFS 또는 DFS 를 이용하여 노드 하나를 시작 노드로 골라서 그 노드와 이어진 그래프를 구한다.

2. 구한 그래프에 포함되지 않은 노드를 다시 시작 노드로 골라서 그래프를 구한다.

3. 2번 절차를 그래프들에 포함되지 않은 노드가 남지 않을 때까지 반복한다.

이를 토대로 구현을 해보았다.

def bfs(computers, start_node):
    queue = list()
    visit = list()
    queue.append(start_node)
    
    while queue:
        node = queue.pop(0)
        
        if node not in visit:
            visit.append(node)
        
        for x in range(0, len(computers[node])):
            if computers[node][x] == 1 and x not in visit:
                queue.append(x)
    
    return visit
            

def solution(n, computers):
    answer = 1
    
    node_list = bfs(computers, 0)
    
    while True:
        for x in range(0, n):
            if x not in node_list:
                new_node = bfs(computers, x)
                node_list.extend(new_node)
                answer += 1
        
        if len(node_list) == n:
            break
            
    return answer

 

저번 BFS 소스코드에 살짝 변화를 주었는데, 여기서는 노드의 번호가 주어지지 않고 0과 1로 이어진 관계만 표시되어 있기 때문이다. 그래서 그것을 노드 번호로 바꿔주기 위한 코드로 변경하였다.

node_list 에서는 그래프에 쓰인 노드를 넣어 놓는 리스트이다. 이 리스트의 길이가 n과 같아질 경우 반복문을 빠져나오게 하였고 리스트에 없는 노드를 시작노드로 설정하게 하였다.

 

문제에 대한 생각

맨 처음 그래프의 갯수를 세야된다는 것을 봤을 때, BFS나 DFS에 대한 생각을 하지 않았다.

하지만 곰곰히 생각을 해보니 그래프 탐색을 하면서 그래프에 쓰이는 노드들을 구할 수 있다는 것을 알았다. 이를 통해 BFS를 사용하니 해답에 다가가는데 크게 어렵지 않았다. DFS는 사용하지 않았지만 DFS도 비슷하게 될 것이라고 생각한다.

 

Written by Sheart