BOJ - 거짓말 1043번 (JAVA)

❓ 문제 - 백준 거짓말 1043번 - JAVA 풀이법

출처

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

1043번: 거짓말

 

 

 

📝 문제해결법

1. 문제

• 지민이가 파티장에 가서 거짓을 얘기하거나 진실을 얘기하게 된다.
• 그러나 어떤 사람들은 지민이가 거짓말을 얘기하는지 알게 된다. 그리고 그 지민이가 거짓말을 얘기하는지, 진실을 얘기하는지 아는 사람들과 같이 파티장에 있는 사람들도 이런 거짓말을 하게 되는지 알게된다.
• 따라서 다른 파티장에서는 거짓말을 한 것을 모르지만, 다른 파티에서는 거짓말을 하는지 알고 있는 파티원과 같이 있으면 모순을 느끼며 거짓말 쟁이로 판단된다.
• 이렇게 모순을 피해 지민이가 거짓말 쟁이가 안되며 최대한 거짓말을 할 수 있는 파티장의 갯수를 구하여라.   

 

2. 해결 방법

• union find(합집합) 알고리즘을 사용하여 문제를 해결한다.
• 우선, 진실인지 거짓인지 아는 파티원들을 boolean true로 관리하며, 각 파티장에 참석한 사람에 대한 정보를 HashSet으로 저장한다.
• 그리고 각 파티장에 대한 정보를 받을 때 각 멤버들을 union find로 해서 같은 집합으로 만든다.
• 그리고 다시 한번, 진실인지 거짓인지 아는 파티원을 기준으로 자신과 같은 집합인 경우인 사람들은 다 진실과 거짓을 판별할 수 있게 되므로 boolean을 true로 변경해준다.
• 그리고 다시 한번, 각 파티장마다 혹시 진실을 아는 사람이 존재한다면 스킵하고 각 파티장에 진실을 아는 사람이 아무도 없을 때 answer 1증가시켜주고, answer를 답으로 출력한다.

 

 

3. 느낀점

• 처음에 union find말고 boolean으로 체킹하면서 비슷하게 로직을 구현했는데, 예제 인풋 아웃풋은 맞았지만.. 틀렸다.. 알고 보니 union find의 합집합으로 진실을 아는 사람 + 진실을 알게 될 사람을 묶고  나중에 다시 한번 더 진실을 알게되는 사람들의 연관관계를 다시 계산해줘야했다.
• union find가 이런 식으로 풀이에 적용될 수 있구나 느꼈다..!

 

 

💻 소스코드 (JAVA)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.HashSet;
import java.util.StringTokenizer;

public class Main_1043 {
    public static int[] parent;
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine(), " ");
        int n = Integer.parseInt(st.nextToken());
        int m = Integer.parseInt(st.nextToken());
        boolean[] know = new boolean[n+1];
        HashSet<Integer>[] party = new HashSet[m+1];
        parent = new int[n+1];
        for(int i=1;i<=m;i++){
            party[i] = new HashSet<>();
        }
        st = new StringTokenizer(br.readLine(), " ");
        int know_cnt = Integer.parseInt(st.nextToken());
        for(int i=1;i<=know_cnt;i++){
            int tmp = Integer.parseInt(st.nextToken());
            know[tmp] = true;
        }

        for(int i=1;i<=n;i++){
            parent[i] = i;
        }

        // 진실을 아는 사람과 같은 파티 공간에 있으면 진실을 아는 것으로 체크
        for(int i=1;i<=m;i++){
            st = new StringTokenizer(br.readLine(), " ");
            int num = Integer.parseInt(st.nextToken());
            if(num <= 0) continue;
            int pre = Integer.parseInt(st.nextToken());
            party[i].add(pre);
            for(int j=0;j<num-1;j++){
                int p = Integer.parseInt(st.nextToken());
                if(find_parent(pre) != find_parent(p)){
                    union(pre, p);
                }

                party[i].add(p);
                pre = p;
            }
        }

        // 진실을 아는 사람과 연관관계가 있음
        boolean[] check = new boolean[n+1];
        for(int i=1;i<=n;i++){
            if(know[i] && !check[i]){
                int p = find_parent(i);
                for(int j=1;j<=n;j++){
                    if(find_parent(j) == p){
                        know[j] = true;
                        check[j] = true;
                    }
                }
            }
        }

        int answer = 0;
        for(int i=1;i<=m;i++){
            boolean c = false;
            for(int p:party[i]){
                if(know[p]){
                    c = true;
                    break;
                }
            }
            if(!c) answer++;
        }

        System.out.println(answer);
    }

    public static int find_parent(int x){
        if(parent[x] == x) return x;
        return parent[x] = find_parent(parent[x]);
    }

    public static void union(int a, int b){
        a = find_parent(a);
        b = find_parent(b);
        if(a <= b){
            parent[b] = a;
        } else {
            parent[a] = b;
        }
    }


}