프로그래머스 Lv3. 네트워크 풀이

https://school.programmers.co.kr/learn/courses/30/lessons/43162

프로그래머스

자바

public class Solution {
    public int solution(int n, int[][] computers) {
        boolean[] visited = new boolean[n];
        int count = 0;

        for (int i = 0; i < n; i++) {
            if (!visited[i]) {
                dfs(computers, visited, i);
                count++;
            }
        }

        return count;
    }

    private void dfs(int[][] computers, boolean[] visited, int node) {
        visited[node] = true;

        for (int i = 0; i < computers.length; i++) {
            if (computers[node][i] == 1 && !visited[i]) {
                dfs(computers, visited, i);
            }
        }
    }
}

 

 

 

파이썬

def solution(n, computers):
    def dfs(node):
        visited[node] = True
        for neighbor in range(n):
            if computers[node][neighbor] == 1 and not visited[neighbor]:
                dfs(neighbor)

    visited = [False] * n
    count = 0

    for i in range(n):
        if not visited[i]:
            dfs(i)
            count += 1

    return count

# Example usage
n = 3
computers = [[1, 1, 0], [1, 1, 0], [0, 0, 1]]
print(solution(n, computers))