[백준 | 실버1] 효율적인 해킹(BFS, 방향성 그래프)

https://www.acmicpc.net/problem/1325

 

문제

해커 김지민은 잘 알려진 어느 회사를 해킹하려고 한다. 이 회사는 N개의 컴퓨터로 이루어져 있다. 김지민은 귀찮기 때문에, 한 번의 해킹으로 여러 개의 컴퓨터를 해킹 할 수 있는 컴퓨터를 해킹하려고 한다.

이 회사의 컴퓨터는 신뢰하는 관계와, 신뢰하지 않는 관계로 이루어져 있는데, A가 B를 신뢰하는 경우에는 B를 해킹하면, A도 해킹할 수 있다는 소리다.

이 회사의 컴퓨터의 신뢰하는 관계가 주어졌을 때, 한 번에 가장 많은 컴퓨터를 해킹할 수 있는 컴퓨터의 번호를 출력하는 프로그램을 작성하시오.

입력

첫째 줄에, N과 M이 들어온다. N은 10,000보다 작거나 같은 자연수, M은 100,000보다 작거나 같은 자연수이다. 둘째 줄부터 M개의 줄에 신뢰하는 관계가 A B와 같은 형식으로 들어오며, "A가 B를 신뢰한다"를 의미한다. 컴퓨터는 1번부터 N번까지 번호가 하나씩 매겨져 있다.

출력

첫째 줄에, 김지민이 한 번에 가장 많은 컴퓨터를 해킹할 수 있는 컴퓨터의 번호를 오름차순으로 출력한다.

예제 입력 1 복사

5 4
3 1
3 2
4 3
5 3

예제 출력 1 복사

1 2

 

 

파이썬

큐를 이용한 방향있는 그래프

import sys
sys.stdin = open("2_4_20220817_백준_1325_효율적인해킹.txt", "r")
from collections import deque

n, m = map(int, input().split())
graph = [[] for _ in range(n + 1)]
countList = [0] * (n+1)

def BFS(node) :
    dq = deque()
    dq.append(node)

    visited = [False] * (n + 1)
    visited[node] = True
    #큐에 있는거 하나씩 빼면서 BFS 탐구
    while dq :
        nowNode = dq.popleft()

        for nextNode in graph[nowNode] :
            if visited[nextNode] is True :
                continue
            else :
                dq.append(nextNode)
                visited[nextNode] = True
    return visited.count(True)


for i in range(m) :
    dstn, src = map(int, input().split())
    graph[src].append(dstn)

for i in range(1, n + 1) :
    countList[i] = BFS(i)
print(countList)
maxNum = max(countList)

for i in range(len(countList)) :
    if countList[i] == maxNum :
        print(i, end=" ")


#n개의 컴퓨터
#A가 B를 신뢰하면, B해킹하면 A 해킹 가능
#한번에 가장 많은 컴퓨터를 해킹할 수 있는 컴퓨터 번호 출력

#그래프 문제
#BFS 문제

 

node.js(자바스크립트)

!!시간초과!!

const fs = require("fs");
const [nm, ...input] = fs
    .readFileSync("input.txt")
    .toString()
    .trim()
    .split("\n");
const [n, m] = nm.split(" ").map(Number);
const array = input.map((el) => el.split(" ").map(Number));

const countList = Array(n + 1).fill(0); //카운트 변수 생성
const graph = Array.from(Array(n + 1), () => []); //빈배열 생성

function BFS(startNode) {
    dq = [];
    dq.push(startNode);

    let visited = Array(n + 1).fill(false);
    visited[startNode] = true;

    while (dq.length) {
        let nowNode = dq.shift();
        //console.log("팝", nowNode);

        for (let nextNode of graph[nowNode]) {
            if (visited[nextNode] === true) {
                continue;
            } else {
                dq.push(nextNode);
                visited[nextNode] = true;
            }
        }
    }

    //배열 내에 있는 true의 개수 찾기 reduce, 초기 count는 0, 찾으면서 개수 +1
    return visited.reduce(
        (count, data) => (data === true ? count + 1 : count),
        0
    );
}

function solution() {
    const result = [];

    for (let i = 0; i < m; i++) {
        const [dstn, src] = array[i];       //a, b입력받기
        graph[src].push(dstn);              //배열에 넣을때는 push
    }

    for (let i = 1; i < n + 1; i++) {
        countList[i] = BFS(i);
    }

    let maxNum = Math.max(...countList);    //최대값 찾을때는 Math.max(...list)

    for (let i = 0; i < countList.length; i++) {
        if (countList[i] === maxNum) {
            result.push(i);
        }
    }

    console.log(result.join(" "));
}

solution();