[Algo Rhythm🕺💃] 프로그래머스 고득점 kit 네트워크

📚 문제 설명

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

 

🎯 어떻게 풀었나

첫번째 시도 (15.4점/ 100점)

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;

class Solution {
    public int solution(int n, int[][] computers) {
        int nNet = n;
        HashMap<Integer, ArrayList<Integer>> graph = new HashMap<>();
        boolean[] visited = new boolean[n];
        
        // init graph
        for(int i = 0; i < computers.length; i++) {
            for(int j = 0; j < computers.length; j++) {
                if(i == j || computers[i][j] != 1) continue;
                
                ArrayList<Integer> lst = (graph.containsKey(i)) ? graph.get(i) : new ArrayList<>();
                lst.add(j);
                graph.put(i, lst);
            }
        }
        
        Iterator<Integer> keys = graph.keySet().iterator();
        while(keys.hasNext()) {
            int k = keys.next();
            visited[k] = true;
            ArrayList<Integer> v = graph.get(k);
            for(int child : v) {
               if(visited[child] == false) {
                    visited[child] = true;
                    nNet--;
                } 
            }
        }
        
        return nNet;
    }
}

 

최종 코드1 (HashMap과 ArrayList를 이용한 graph표현)

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;

class Solution {
    public int solution(int n, int[][] computers) {
        int nNet = 0;
        HashMap<Integer, ArrayList<Integer>> graph = new HashMap<>();
        boolean[] visited = new boolean[n];
        
        // init graph
        for(int i = 0; i < computers.length; i++) {
            ArrayList<Integer> lst = new ArrayList<>();
            for(int j = 0; j < computers.length; j++) {
                if(i == j || computers[i][j] != 1) continue;
                
                lst.add(j);
            }
            graph.put(i, lst);
        }
        
        Iterator<Integer> keys = graph.keySet().iterator();
        while(keys.hasNext()) {
            int k = keys.next();
            if(visited[k] == false) {
                dfs(graph, visited, k);
                nNet++;
            }
        }
        
        return nNet;
    }
    
    public void dfs(HashMap<Integer, ArrayList<Integer>> graph, boolean[] visited, int parent) {
        if(visited[parent] == true) return;
        
        visited[parent] = true;
        for(int child : graph.get(parent)) {
         dfs(graph, visited, child);   
        }
    }
}

 

최종 코드2 (ArrayList를 이용한 graph 표현)

import java.util.ArrayList;

class Solution {
    public int solution(int n, int[][] computers) {
        int nNet = 0;
        ArrayList<ArrayList<Integer>> graph = new ArrayList<>();
        boolean[] visited = new boolean[n];
        
        // init graph
        for(int i = 0; i < computers.length; i++) {
            ArrayList<Integer> children = new ArrayList<>();
            for(int j = 0; j < computers.length; j++) {
                if(i == j || computers[i][j] != 1) continue;
                children.add(j);
            }
            graph.add(children);
        }
        
        for(int i = 0; i < graph.size(); i++) {
            if(visited[i] == false) {
                dfs(graph, visited, i);
                nNet++;
            }
        }
        
        return nNet;
    }
    
    public void dfs(ArrayList<ArrayList<Integer>> graph, boolean[] visited, int parent) {
        if(visited[parent] == true) return;
        
        visited[parent] = true;
        for(int child : graph.get(parent)) {
            dfs(graph, visited, child);
        }
    }
}

 

🧐 왜 저렇게 풀었나

첫번째 시도(15.4점/ 100점)

처음에는 네트워크가 컴퓨터의 개수만큼 있고 한 컴퓨터와 다른 컴퓨터가 네트워크로 이어져 있으면 이를 하나로 취급하도록 코드를 작성했다. DFS나 BFS 둘 중 아무것도 사용하지 않았다. 새로운 풀이법을 발견한 것인가하며 기대했지만 결과는 대실패였다! 🤦‍♀️🤣

 

최종 코드1(HashMap과 ArrayList를 이용한 graph 표현), 최종 코드 2(ArrayList를 이용한 graph 표현)

알고리즘은 복잡하지 않다. 일단 그래프를 만든다. 그리고나서 노드 하나씩 살펴보는데 만약 노드가 이전에 방문한 적이 없다면 이어진 모든 노드들을 방문하고 네트워크 개수를 하나 늘려준다.

 

📖 무엇을 배웠나

DFS 생각하는 거 너무 재밌다 😆😆