[백준 C++] 1034 거짓말

문제

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

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

입력

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

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

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

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

출력

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

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

풀이

맨처음에는 M개의 파티구성에대한 입력이 주어질때마다 입력받아서 그때그때 DSU알고리즘으로 지민이가 거짓말이 가능한지 파악했더니 틀렸다.
문제를 보면

당연히, 어떤 사람이 어떤 파티에서는 진실을 듣고, 또다른 파티에서는 과장된 이야기를 들었을 때도 지민이는 거짓말쟁이로 알려지게 된다.

이 말은 즉, 모든 파티구성에 대한 입력을 다받은뒤 전부 DSU과정을 거친 후에,
지민이가 다시 한번 모든 파티목록을 탐색하며 거짓말이 가능한지 파악해야한다는소리.
즉 M번의 파티중 k번째 파티에서 진실을 아는사람이 없어서 지민이가 거짓말을 했더라도 k번째 파티구성원이
k+1, k+2, ''' M번째 파티에서 진실을 아는 사람을 만나면,
k번째에서 지민이가 거짓말을 했으면 안되게 된다는소리.

즉 두번 탐색해야한다.

#define _CRT_SECURE_NO_WARNINGS 
#include <bits/stdc++.h>
using std::vector;
int N, M, T;
int parent[52]; //1~51
vector<int> partys[51]; //1~M

int res = 0; //가능한 파티수
int* truth;
void goToParty();
void init();
void Union(int x, int y);
int Find(int x);

int Find(int x) {
    if (parent[x] == x)
        return x;
    return parent[x] = Find(parent[x]); //경로압축
}

void Union(int x, int y) {
    x = Find(x);
    y = Find(y);
    if (x > y)
        parent[y] = x; //더큰 부모노드에 자식으로 삽입
    else
        parent[x] = y;
}

void init() {
    scanf("%d%d%d", &N, &M, &T);
    for (int i = 1; i <= N; i++) //초기화
        parent[i] = i;
    parent[51] = 51;
    truth = new int[T];

    for (int i = 0; i < T; i++)
        scanf("%d", &truth[i]);

    if (T == 0) //아래의 코드가 오류나는것을 방지
        return;

    
    for (int i = 0; i < T; i++)
        parent[truth[i]] = 51; //진실을 말하는 사람은 모두 부모노드로 51지목

}
//M개의 파티목록을 입력받고 각 파티별 Union연산을 실행
void goToParty() { 
    for (int k = 1; k <= M; k++) {
        int num;
        scanf("%d", &num);
        bool canLie = true;

        for (int i = 0; i < num; i++) {
            int human;
            scanf("%d", &human);
            partys[k].push_back(human); //k번째 파티목록에 저장
        }
        //진실을 아는사람이 없으면 모든 파티에서 거짓말가능
        //앞서 입력까지만 받고 그 뒤 과정은 모두 생략하도록
        if (T == 0) {
            res++;
            continue;
        }

        for (int i = 0; i < num - 1; i++)
            Union(partys[k][i], partys[k][i + 1]); //파티 참여자끼리 합연산
    }
}

//M개의 파티구성원 목록중 거짓말이 가능한 경우를 계산
void calculate() {
    if (T == 0) {
        printf("%d", M); //모든 파티에서 거짓말 가능
        return;
    }
    
    for (int k = 1; k <= M; k++) {
        int size = partys[k].size();
        bool canLie = true;
        for (int i = 0; i < size; i++) {
            //k번째 파티의 i번 참가자의 부모노드가 
            //51(진실을 아는 사람그룹의 루트노드번호)이라면
            if (Find(partys[k][i]) == 51) {
                canLie = false; //이 파티에서 지민이는 거짓말을 할수없음
                break;
            }
        }

        if (canLie)
            res++;
    }

    printf("%d", res);
}

int main(void) {
    init();
    goToParty();
    calculate();

    return 0;
}