Programmers Code/Level 3_4

[Programmers] level3 - 네트워크 (Python) : DFS/BFS

NIMHO 2022. 6. 10. 13:08
728x90

▶네트워크 - DFS/BFS (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

 

풀이

이 문제는 DFS를 이용해서 풀었다.

연결된 노드를 다 확인을 해주고 visited로 방문 유무 또한 체크해주었다.

solution함수에서 visited가 -1일때는 또다서 dfs를 돌려주면서 풀었다.

 

def solution(n, computers):
    answer = 0
    visited = [-1 for i in range(n)]
    
    for com in range(n):
        if visited[com] == -1:
            DFS(n, computers, com, visited)
            answer += 1
    return answer


def DFS(n, computers, com, visited):
    visited[com] = 0
    for connect in range(n):
        if connect != com and computers[com][connect] == 1:
            if visited[connect] == -1:
                DFS(n, computers, connect, visited)

효율성 테스트는 없었고, 정확성 테스트만 통과하면 된다.

효율성 테스트가 없다보니 조금 더 쉽게 해결할 수 있었던거 같다.

 

정확성 테스트

테스트 1 통과 (0.01ms, 10.2MB)
테스트 2 통과 (0.01ms, 10.2MB)
테스트 3 통과 (0.07ms, 10.2MB)
테스트 4 통과 (0.06ms, 10.2MB)
테스트 5 통과 (0.01ms, 10.2MB)
테스트 6 통과 (0.31ms, 10.2MB)
테스트 7 통과 (0.02ms, 10.2MB)
테스트 8 통과 (0.23ms, 10.2MB)
테스트 9 통과 (0.13ms, 10.2MB)
테스트 10 통과 (0.15ms, 10.2MB)
테스트 11 통과 (0.99ms, 10.2MB)
테스트 12 통과 (0.86ms, 10.2MB)
테스트 13 통과 (0.45ms, 10.2MB)
채점 결과
정확성: 100.0
합계: 100.0 / 100.0
728x90