[C#] 프로그래머스 문제 - 네트워크(DFS)

문제 설명

네트워크란 컴퓨터 상호 간에 정보를 교환할 수 있도록 연결된 형태를 의미합니다. 예를 들어, 컴퓨터 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개의 네트워크가 있습니다.

 

 

내 풀이

using System;
using System.Collections.Generic;
using System.Linq;
public class Solution {
    public int solution(int n, int[,] computers) {
        int answer = 1;
        Stack<int> node = new Stack<int>();
        HashSet<int> unvisited =new HashSet<int>();
        //아직 탐색 전에 모든 노드는 비 방문 상태.
        for(int i=0; i<n; i++) unvisited.Add(i);
        node.Push(0);
        while(unvisited.Count > 0)
        {
            if(node.Count ==0)
            {
                foreach(var i in unvisited)
                {
                    node.Push(i);
                    answer++;
                    break;
                }
            }
            int currentnode = node.Pop();
            unvisited.Remove(currentnode);
            for(int i=0; i<n; i++)
            {
                //만약 노드 자기 자신 값이면 패스
                if(i==currentnode) continue;
                //만약 i노드가 unvisted가 아니라면 패스
                if(!unvisited.Contains(i)) continue;
                //자식 노드 푸쉬
                if(computers[currentnode,i]==1) node.Push(i);
            }
        }
        return answer;
    }
}

 

풀이 

 

1. 내가 방문할 노드를 스택으로 구현한다(DFS)

 

2. 내가 아직 방문하지 않은 노드를 HashSet으로 구현해 모든 노드를 넣어둔다.

 

3. 루트 노드를 넣어 준다. A번째 노드(맨처음 노드이니 0으로)

 

3. 첫번째 반복문 조건은 아직 방문하지 않은 노드가 남았다면 반복한다.

 

4. 만약 아직 방문하지 않은 노드가 남았는데 스택이 비었다면 스택에 방문하지 않은 노드를 넣어준다.

 이때 answer 값도 추가됨. (네트워크가 추가되는 시점.)

 

5. 자식 노드를 추가해주는 조건은 노드 자기 자신이 아니고 이미 방문했던 노드가 아닌 노드를 발견하였을 때 추가해준다.

 

위 조건을 모두 지키면 네트워크의 개수가 return된다.

 

느낀 점: 사실 프로그래머스 사이트에 DFS&BFS 문제로 분류되어 있지만 모든 노드를 방문해야 한다는 점에서 해당 방법이 완벽히 적합한 방법이라고 느껴지지 않았다. 스택이 아닌 큐로 문제를 풀어 BFS로 풀어도 수행 속도의 차이가 크게 존재하지 않았을 것 같다.