hongjunland

Welcome

2 분 소요

백준 1043

제한

시간제한 메모리 제한 제출 정답 맞은 사람 정답 비율
2 초 128 MB 9335 3569 2831 40.687%

문제

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

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

입력

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

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

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

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

출력

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

예제 입력1

4 3
0
2 1 2
1 3
3 2 3 4

예제 출력1

3

예제 입력2

4 1
1 1
4 1 2 3 4

예제 출력2

0

예제 입력3

4 1
0
4 1 2 3 4

예제 출력3

1

풀이

진실을 아는 사람들의 집합에 대한 알고리즘이므로, 분리집합(Disjoint Set) 알고리즘을 사용하여 구현하였다.

진실을 아는사람에 대한 첫번째 원소를 루트노드로 설정하고 같은 집합에 포함시킨다.

파티리스트를 순회하면서, 인원중의 루트노드가 진실을 아는 사람의 루트와 같다면 해당 인원을 모두 포함시칸다.

그리고 다시 순회하면서, 진실을 아는사람의 집합에 포함 되어 있지않는 경우에만 카운팅을해서 출력한다.

소스코드


import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.StringTokenizer;

public class Main {
    static int N,M;
    static int T;
    static int[] parents;   // 부모
    static int host = -1;   // 진실을 아는집합의 루트 인덱스   
    static List[] parties;  // 파티 예정 목록  
    static int Result = 0;
    static void make(){     // 부모인덱스 초기화
        for(int i = 1 ; i <= N ; i++){
            parents[i] = i;
        }
    }
    static int find(int a){
        if(parents[a]==a) return a;
        return parents[a] = find(parents[a]);
    }
    static boolean union(int a, int b){
        int aRoot = find(a);
        int bRoot = find(b);
        if(aRoot==bRoot) return false;
        parents[bRoot] = aRoot;
        return true;
    }
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        N = Integer.parseInt(st.nextToken());
        M = Integer.parseInt(st.nextToken());
        parents = new int[N+1];
        parties = new ArrayList[M];
        make();
        st = new StringTokenizer(br.readLine());
        T = Integer.parseInt(st.nextToken());

        for(int i = 0 ; i < T ; i++){
            int person = Integer.parseInt(st.nextToken());
            if(host==-1){   // 가장 빠른 인덱스가 루트 인덱스
                host = person;
            }else{
                union(host,person);
            }
        }
        for(int i = 0 ; i < M ; i++){
            st = new StringTokenizer(br.readLine());
            int size = Integer.parseInt(st.nextToken());
            parties[i] = new ArrayList<Integer>();
            int head = Integer.parseInt(st.nextToken());
            parties[i].add(head);
            for(int j = 1 ; j < size ; j++){
                int person = Integer.parseInt(st.nextToken());
                parties[i].add(person);
                union(head,person);
            }
        }

        for (int i = 0 ; i < M ; i++){
            List<Integer> persons = parties[i];
            boolean flag = false;
            if(host==-1) {
                Result++;
                continue;
            }
            for(int person : persons){
                if(find(person)==find(host)){
                    flag = true;
                    break;
                }
            }
            if(!flag) Result++;
        }
        System.out.println(Result);
    }

}

참고