[BoJ] - 1043번 거짓말

문제를 풀면서 1시간 정도 고민을 했었고  1시간 구현해서 풀어낸 문제이다. 골드 4이지만 난이도가 막 어렵지는 않았던 것 같다. 오랜만에 골드 4를 스스로의 힘으로 풀어서 기분이 좋았다

 

문제

---

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

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

 

입력

---

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

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

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

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

 

 

처음에 고민을 했던건 Map으로 거짓말을 아는 사람을 넣어두고 파티에 같이 참여한 사람들을 동적으로 넣어주려고 하다가 Map은 for문을 돌때 key 값을 동적으로 추가를 못한다는걸 알고 큐로 급하게 구현을 했던것 같다. 

 

우선 이차원 배열 graph 를 선언해두고 row에는 파티 순서 column 에는 파티에 참여한 사람 no 을 부여하여 해당 파티에 참여한 사람들의 no를 true로 변경을 해주었다 

 

그리고 구현부에서는 graph를 순환하면서 거짓말을 아는사람의 포함된 파티에 참여자를 queue에 넣어 다른 파티에도 참여한 이력이 있으면 party라는 배열을 두어 true로 counting 하는데 제거하였다 

// 거짓말을 아는사람이 없을때 까지 돌리기 
while(!knowq.isEmpty()) {
    int key = knowq.poll();
    // 거짓말을 아는사람의 참여한 파티를 Loop 돌기 
    for (int i = 1; i <= m; i++) {
    	// 거짓말을 아는사람이 참여한 파티라면
        if(graph[i][key]) {
        	// 해당 파티는 true로 표기하여 이후 counting 할때 제외
            partys[i] = true;
            // 그 파티에 참여한 사람을 no으로 loop 
            for (int j = 1; j <= n; j++) {
            	// knowq에 방문한 적이 없고 그파티에 참여한 사람이라면 
                if(!visited[j] && graph[i][j]) {
                	// 거짓말을 아는 사람으로 확인되어 queue 에 넣는다
                    knowq.add(j);
                    // 거짓말을 아는사람 방문자 처리 
                    visited[j] = true;
            	}
            }
        }
    }
}

 

전체 코드는 아래와 같다

import java.io.*;
import java.util.*;

class 거짓말{
    static BufferedReader br;
    static StringBuilder sb;
    static StringTokenizer st;
    static int n,m;
    static Queue<Integer> knowq;
    static boolean[] partys;
    static boolean[][] graph;
    static boolean[] visited;
    // input
    static void input () {
        try {
            // 라이브
            // br = new BufferedReader(new InputStreamReader(System.in));
            // 개발 테스트
            br = new BufferedReader(new FileReader("test.txt"));
            sb = new StringBuilder();
            st = new StringTokenizer(br.readLine()," ");
            n = parseInt(st.nextToken());
            m = parseInt(st.nextToken());
            graph = new boolean[m+1][n+1];
            visited = new boolean[n+1];
            knowq = new LinkedList<>();
            st = new StringTokenizer(br.readLine(), " ");
            int knowNum = parseInt(st.nextToken());
            partys = new boolean[m+1];
            // know Setting
            for (int i = 0; i < knowNum; i++) {
                int num = parseInt(st.nextToken());
                knowq.add(num);
                visited[num] = true;
            }
            for (int i = 1; i <= m; i++) {
                st = new StringTokenizer(br.readLine(), " ");
                int partyNum = parseInt(st.nextToken());
                for (int j = 0; j < partyNum; j++) {
                    int no = parseInt(st.nextToken());
                    graph[i][no]  = true;
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        }

    }
    //main
    public static void main(String[] args) throws IOException {
        input();
        while(!knowq.isEmpty()) {
            int key = knowq.poll();
            for (int i = 1; i <= m; i++) {
                if(graph[i][key]) {
                    partys[i] = true;
                    for (int j = 1; j <= n; j++) {
                        if(!visited[j] && graph[i][j]) {
                            knowq.add(j);
                            visited[j] = true;
                        }
                    }
                }
            }
        }
        int cnt = 0;
        for (int i = 1; i < partys.length; i++) {
            if(!partys[i]) {
                cnt += 1;
            }
        }
        System.out.println(cnt);
    }
    private static int parseInt(String str) {
        return Integer.parseInt(str);
    }
}