[백준] 1043 : 거짓말 https://www.acmicpc.net/problem/1043
문제
지민이는 파티에 가서 이야기 하는 것을 좋아한다. 파티에 갈 때마다, 지민이는 지민이가 가장 좋아하는 이야기를 한다. 지민이는 그 이야기를 말할 때, 있는 그대로 진실로 말하거나 엄청나게 과장해서 말한다. 당연히 과장해서 이야기하는 것이 훨씬 더 재미있기 때문에, 되도록이면 과장해서 이야기하려고 한다. 하지만, 지민이는 거짓말쟁이로 알려지기는 싫어한다. 문제는 몇몇 사람들은 그 이야기의 진실을 안다는 것이다. 따라서 이런 사람들이 파티에 왔을 때는, 지민이는 진실을 이야기할 수 밖에 없다. 당연히, 어떤 사람이 어떤 파티에서는 진실을 듣고, 또다른 파티에서는 과장된 이야기를 들었을 때도 지민이는 거짓말쟁이로 알려지게 된다. 지민이는 이런 일을 모두 피해야 한다.
사람의 수 N이 주어진다. 그리고 그 이야기의 진실을 아는 사람이 주어진다. 그리고 각 파티에 오는 사람들의 번호가 주어진다. 지민이는 모든 파티에 참가해야 한다. 이때, 지민이가 거짓말쟁이로 알려지지 않으면서, 과장된 이야기를 할 수 있는 파티 개수의 최댓값을 구하는 프로그램을 작성하시오.
입력
첫째 줄에 사람의 수 N과 파티의 수 M이 주어진다.
둘째 줄에는 이야기의 진실을 아는 사람의 수와 번호가 주어진다. 진실을 아는 사람의 수가 먼저 주어지고 그 개수만큼 사람들의 번호가 주어진다. 사람들의 번호는 1부터 N까지의 수로 주어진다.
셋째 줄부터 M개의 줄에는 각 파티마다 오는 사람의 수와 번호가 같은 방식으로 주어진다.
N, M은 50 이하의 자연수이고, 진실을 아는 사람의 수는 0 이상 50 이하의 정수, 각 파티마다 오는 사람의 수는 1 이상 50 이하의 정수이다.
출력
첫째 줄에 문제의 정답을 출력한다.
문제 해석
진실을 알고 있는 사람과 한번이라도 같은 파티에 참석한 적이 있다면, 진실을 아는 사람이 된다.
진실을 아는 사람이 한 명도 없는 파티에서만 지민이가 과장해서 얘기할 수 있다.
즉, 진실을 아는 사람이 한 명도 없는 파티의 수를 출력해야 한다.
풀이
boolean 타입의 2차원 배열을 이용하여 그래프를 표현했다.
파티의 참석자가 혼자라면, 참석자의 번호에 해당하는 곳(번호, 번호)을 true로 바꾼다.
파티의 참석자가 여러명이라면, 순차적으로 연결해서 해당하는 곳(앞사람 번호, 뒷사람 번호)을 true로 바꾼다.
만약, 진실을 알고 있는 사람과 같은 행 또는 같은 열에 있는 사람이 있다면 그 사람은 진실을 알게 된다.
재귀적으로 진실을 알고 있는 사람과 같은 행 또는 열에 있는 사람을 false로 바꾼다.
최종적으로 파티의 참석자들을 순회하며 모두 true인 경우 count를 올려준다.
순회가 끝난 후 count값을 출력한다.
코드
import java.io.*;
public class Main {
static int N;
static int M;
static int[] liar;
static boolean[][] graph;
static int[][] member;
static int count;
static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
static BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
static void input() throws IOException {
String[] input = br.readLine().split(" ");
N = Integer.parseInt(input[0]);
M = Integer.parseInt(input[1]);
input = br.readLine().split(" ");
int n = Integer.parseInt(input[0]);
liar = new int[n];
graph = new boolean[N+1][N+1];
member = new int[M][51];
for(int i=0; i<n; i++) {
liar[i] = Integer.parseInt(input[i+1]);
}
for(int i=0; i<M; i++) {
input = br.readLine().split(" ");
for(int k=0; k<input.length; k++) {
member[i][k] = Integer.parseInt(input[k]);
}
int m = Integer.parseInt(input[0]);
if(m == 1) {
int index = Integer.parseInt(input[1]);
graph[index][index] = true;
} else {
for(int j=1; j<m; j++) {
graph[Integer.parseInt(input[j])][Integer.parseInt(input[j+1])] = true;
}
}
}
}
static void delete() {
for(int index : liar) {
makeFalse(index);
}
}
static void makeFalse(int index) {
for(int i=1; i<=N; i++) {
if(graph[index][i] == true) {
graph[index][i] = false;
makeFalse(i);
}
if(graph[i][index] == true) {
graph[i][index] = false;
makeFalse(i);
}
}
}
static void result() throws IOException {
count = 0;
boolean check;
for(int i=0; i<M; i++) {
check = true;
if(member[i][0] == 1) {
int n = member[i][1];
if(graph[n][n]) {
count++;
}
check = false;
} else {
for(int j=1; j<member[i][0]; j++) {
if(!graph[member[i][j]][member[i][j+1]]) {
check = false;
break;
}
}
}
if(check) {
count++;
}
}
bw.write(String.valueOf(count));
}
public static void main(String[] args) throws IOException {
input();
delete();
result();
bw.flush();
bw.close(); br.close();
}
}
결과
후기
좀 더 발전적인 방향이 있을 것 같지만, 메모리와 시간소요가 생각보다 적어 해당 방식으로 제출했다!
최대한 중복된 행위를 지양하는 코드를 짜도록 해야겠다,,
수정할 사항이 있다면 댓글 부탁드립니다 :)
