1043 거짓말 (자바스크립트)

문제

지민이는 파티에 가서 이야기 하는 것을 좋아한다. 파티에 갈 때마다, 지민이는 지민이가 가장 좋아하는 이야기를 한다. 지민이는 그 이야기를 말할 때, 있는 그대로 진실로 말하거나 엄청나게 과장해서 말한다. 당연히 과장해서 이야기하는 것이 훨씬 더 재미있기 때문에, 되도록이면 과장해서 이야기하려고 한다. 하지만, 지민이는 거짓말쟁이로 알려지기는 싫어한다. 문제는 몇몇 사람들은 그 이야기의 진실을 안다는 것이다. 따라서 이런 사람들이 파티에 왔을 때는, 지민이는 진실을 이야기할 수 밖에 없다. 당연히, 어떤 사람이 어떤 파티에서는 진실을 듣고, 또다른 파티에서는 과장된 이야기를 들었을 때도 지민이는 거짓말쟁이로 알려지게 된다. 지민이는 이런 일을 모두 피해야 한다.

사람의 수 N이 주어진다. 그리고 그 이야기의 진실을 아는 사람이 주어진다. 그리고 각 파티에 오는 사람들의 번호가 주어진다. 지민이는 모든 파티에 참가해야 한다. 이때, 지민이가 거짓말쟁이로 알려지지 않으면서, 과장된 이야기를 할 수 있는 파티 개수의 최댓값을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 사람의 수 N과 파티의 수 M이 주어진다.

둘째 줄에는 이야기의 진실을 아는 사람의 수와 번호가 주어진다. 진실을 아는 사람의 수가 먼저 주어지고 그 개수만큼 사람들의 번호가 주어진다. 사람들의 번호는 1부터 N까지의 수로 주어진다.

셋째 줄부터 M개의 줄에는 각 파티마다 오는 사람의 수와 번호가 같은 방식으로 주어진다.

N, M은 50 이하의 자연수이고, 진실을 아는 사람의 수는 0 이상 50 이하의 정수, 각 파티마다 오는 사람의 수는 1 이상 50 이하의 정수이다.

출력

첫째 줄에 문제의 정답을 출력한다.

====

문제 풀이

우선 진실을 아는 사람이 없다면 정답은 파티의 갯수 그대로 출력 되어야한다.

문제에서 진실을 알고 있는 사람과 같은 파티에 있는 사람들은 진실임을 알게 되어 거짓말이 통하지 않는다고 한다.

이를 통해 진실을 알고 있는 사람부터 시작하여 진실 전염을 시켜서 거짓말이 통하지 않는 사람들을 모두 체크해줘야 한다.

그러기 위해선 사람들의 네트워크를 알아야한다. 같은 파티에 있다면 이 사람들끼리는 연결이 되어 있다는 것이므로 파티를 체크하면서 네트워크 형성을 해준다. 이때 중복을 피하기 위해 각 사람들에 대해 set을 선언해준 뒤 add 해줌.

peoples는 사람들의 네트워크를 기록하기 위함.

isTrues는 진실을 알게 되는 사람인지를 체크하기 위한 배열

const input = require('fs')
    .readFileSync(process.platform === 'linux' ? '/dev/stdin' : './input.txt')
    .toString()
    .trim()
    .split('\n');

const [n, m] = input[0].split(' ').map(Number);
const [trueCnt, ...truePeople] = input[1].split(' ').map(Number);
if (trueCnt === 0) {
    console.log(m);
    process.exit(0);
}

const peoples = Array.from({ length: n + 1 }, () => new Set());
const isTrues = Array.from({ length: n + 1 }, () => false);

for (let i = 2; i < m + 2; i++) {
    const [peopleCnt, ...partiesPeople] = input[i].split(' ').map(Number);

    for (let j = 0; j < partiesPeople.length - 1; j++) {
        const s = partiesPeople[j];
        for (let k = j + 1; k < partiesPeople.length; k++) {
            const e = partiesPeople[k];
            peoples[s].add(e);
            peoples[e].add(s);
        }
    }
}

아래는 진실을 알고 있는 초기 사람들로부터 시작하여 BFS 방식으로 진실을 감염시키는 거다

for (const start of truePeople) {
    const q = [start];
    isTrues[start] = true;
    while (q.length) {
        const cur = q.shift();
        for (const next of peoples[cur]) {
            if (isTrues[next]) continue;
            isTrues[next] = true;
            q.push(next);
        }
    }
}

이제 각 파티를 다시 체크해가면서 진실을 알고 있는 사람이 한 사람이라도 있다면 건너뛰고 아니라면 answer에 1씩 더해준다.

let answer = 0;
for (let i = 2; i < m + 2; i++) {
    const [peopleCnt, ...partiesPeople] = input[i].split(' ').map(Number);

    if (partiesPeople.some((v) => isTrues[v])) continue;
    answer++;
}
console.log(answer);

정답 코드

const input = require('fs')
    .readFileSync(process.platform === 'linux' ? '/dev/stdin' : './input.txt')
    .toString()
    .trim()
    .split('\n');

const [n, m] = input[0].split(' ').map(Number);
const [trueCnt, ...truePeople] = input[1].split(' ').map(Number);
if (trueCnt === 0) {
    console.log(m);
    process.exit(0);
}

const peoples = Array.from({ length: n + 1 }, () => new Set());
const isTrues = Array.from({ length: n + 1 }, () => false);

for (let i = 2; i < m + 2; i++) {
    const [peopleCnt, ...partiesPeople] = input[i].split(' ').map(Number);

    for (let j = 0; j < partiesPeople.length - 1; j++) {
        const s = partiesPeople[j];
        for (let k = j + 1; k < partiesPeople.length; k++) {
            const e = partiesPeople[k];
            peoples[s].add(e);
            peoples[e].add(s);
        }
    }
}

for (const start of truePeople) {
    const q = [start];
    isTrues[start] = true;
    while (q.length) {
        const cur = q.shift();
        for (const next of peoples[cur]) {
            if (isTrues[next]) continue;
            isTrues[next] = true;
            q.push(next);
        }
    }
}

let answer = 0;
for (let i = 2; i < m + 2; i++) {
    const [peopleCnt, ...partiesPeople] = input[i].split(' ').map(Number);

    if (partiesPeople.some((v) => isTrues[v])) continue;
    answer++;
}
console.log(answer);