[백준] 1043

거짓말

문제

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

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

입력

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

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

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

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

출력

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

풀이

#include <iostream>
#include <vector>
#include <queue>

int arr[51][51] = {0};
std::vector<int> parties[51];
int visited[51];
std::queue<int> que;

void bfs(int n) {
    while (!que.empty()) {
        int current = que.front();
        visited[current] = 1;
        que.pop();
        for (int i = 1; i <= n; i++) {
            if (arr[i][current] == 1 && visited[i] == 0) {
                visited[i] = 1;
                que.push(i);
            }
        }
    }
}

int main(void) {
    int n, m, l, temp;
    int count = 0;

    std::cin >> n >> m >> l;
    for (int i = 0; i < l; i++) {
        std::cin >> temp;
        que.push(temp);
    }

    for (int i = 0; i < m; i++) {
        std::cin >> l;
        for (int j = 0; j < l; j++) {
            std::cin >> temp;
            parties[i].push_back(temp);
        }

        for (int j = 0; j < l; j++) {
            if (j == l - 1) continue;
            arr[parties[i][j]][parties[i][j + 1]] = 1;
            arr[parties[i][j + 1]][parties[i][j]] = 1;
        }
    }

    bfs(n);

    for (int i = 0; i < m; i++) {
        for (std::vector<int>::iterator it = parties[i].begin(); it != parties[i].end(); it++) {
            if (visited[*it] == 1) {
                count++;
                break;
            }
        }
    }
    std::cout << m - count;
}

반성회

1. BFS에서 사용할 큐에 진실을 아는 사람들을 미리 넣어놓는다
2. 파티 벡터 배열 (parties) 에 파티원들을 담는다
3. 간선을 나타내는 배열 (arr) 에서 연결되어 있는 (같은 파티에 속한) 사람들끼리 1로 잇는다
4. 큐가 빌 때까지 (진입 지점이 소진될 때까지) bfs를 돌린다
   1. 큐의 맨 앞에서 current를 가져온다
   2. current는 진실을 아는 사람임이 자명하니 visited[current] 를 1로 만든다
   3. current 를 pop한다
   4. 반복문을 돌면서 n명의 사람들에 대해 current와 연결되어 있는지, 그리고 visited가 0인지 확인한다
   5. current와 연결되어 있으면서 visited가 0인 경우, current를 통해 진실을 알아버린 사람이므로 visited를 1로 만들고 큐에 push한다
5. 파티를 돌면서 한 명이라도 진실을 아는 자 (visited 배열에서의 값이 1인 사람) 가 있을 경우 해당 파티의 개수를 센다
6. 전체 파티에서 진실을 아는 파티의 개수를 빼서 진실을 모르는 파티의 개수를 센다

파티에 가서 한 번이라도 마주친 애들끼리 간선으로 연결 후, BFS를 이용하여 진실을 아는 이들을 퍼트려나갔다

마지막엔 진실을 아는 사람이 한 명이라도 포함된 파티 수를 센 뒤, 전체 파티 수에서 빼주었다

입력값을 받아와서 간선에 넣어주는 게 좀 빡셌다…