문제 설명
지민이는 파티에 가서 이야기 하는 것을 좋아한다. 파티에 갈 때마다, 지민이는 지민이가 가장 좋아하는 이야기를 한다. 지민이는 그 이야기를 말할 때, 있는 그대로 진실로 말하거나 엄청나게 과장해서 말한다. 당연히 과장해서 이야기하는 것이 훨씬 더 재미있기 때문에, 되도록이면 과장해서 이야기하려고 한다. 하지만, 지민이는 거짓말쟁이로 알려지기는 싫어한다. 문제는 몇몇 사람들은 그 이야기의 진실을 안다는 것이다. 따라서 이런 사람들이 파티에 왔을 때는, 지민이는 진실을 이야기할 수 밖에 없다. 당연히, 어떤 사람이 어떤 파티에서는 진실을 듣고, 또다른 파티에서는 과장된 이야기를 들었을 때도 지민이는 거짓말쟁이로 알려지게 된다. 지민이는 이런 일을 모두 피해야 한다.
사람의 수 N이 주어진다. 그리고 그 이야기의 진실을 아는 사람이 주어진다. 그리고 각 파티에 오는 사람들의 번호가 주어진다. 지민이는 모든 파티에 참가해야 한다. 이때, 지민이가 거짓말쟁이로 알려지지 않으면서, 과장된 이야기를 할 수 있는 파티 개수의 최댓값을 구하는 프로그램을 작성하시오.
입력
첫째 줄에 사람의 수 N과 파티의 수 M이 주어진다.
둘째 줄에는 이야기의 진실을 아는 사람의 수와 번호가 주어진다. 진실을 아는 사람의 수가 먼저 주어지고 그 개수만큼 사람들의 번호가 주어진다. 사람들의 번호는 1부터 N까지의 수로 주어진다.
셋째 줄부터 M개의 줄에는 각 파티마다 오는 사람의 수와 번호가 같은 방식으로 주어진다.
N, M은 50 이하의 자연수이고, 진실을 아는 사람의 수와 각 파티마다 오는 사람의 수는 모두 0 이상 50 이하의 정수이다.
출력
첫째 줄에 문제의 정답을 출력한다.
문제 풀이
- 진실을 아는 사람들의 번호를 입력 받아 큐에 저장한다.
- 같은 파티에 참석한 사람들의 인접 관계를 저장한다.
- 같은 파티에 참석하면 같은 이야기를 듣게 되므로 진실을 아는 사람과 파티에서 만난 사람은 진실을 알게 되는 점을 이용하여 인접 관계와 큐를 이용하여 진실을 아는 사람의 목록을 갱신한다.
- 지민이가 거짓말을 할 수 있는 파티 개수를 구한다.
(파티에 진실을 아는 사람이 아무도 참석하지 않아야 함)
소스 코드
public static void main(String[] args) throws NumberFormatException, IOException {
BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(in.readLine(), " ");
int N = Integer.parseInt(st.nextToken());
int M = Integer.parseInt(st.nextToken());
// 진실을 아는 사람의 수와 번호
st = new StringTokenizer(in.readLine(), " ");
int T = Integer.parseInt(st.nextToken());
boolean truth[] = new boolean[N+1];
Queue<Integer> queue = new LinkedList<Integer>();
for (int i = 0; i < T; i++) {
int tmp = Integer.parseInt(st.nextToken());
truth[tmp] = true;
queue.add(tmp);
}
// 같은 파티에 참석한 사람들
Set<Integer> adj[] = new HashSet[N+1];
for (int i = 1; i <= N; i++) {
adj[i] = new HashSet<Integer>();
}
// 파티에 오는 사람의 수와 번호
int party[][] = new int[M][N];
for (int i = 0; i < M; i++) {
st = new StringTokenizer(in.readLine(), " ");
int len = Integer.parseInt(st.nextToken());
for (int j = 0; j < len; j++) {
int tmp = Integer.parseInt(st.nextToken());
party[i][j] = tmp;
// 같은 파티에 참석한 사람들 목록에 추가
for (int k = 0; k < j; k++) {
adj[tmp].add(party[i][k]);
adj[party[i][k]].add(tmp);
}
}
}
// 진실을 아는(들은) 사람 갱신
while (!queue.isEmpty()) {
int tmp = queue.poll();
for (int i : adj[tmp]) {
// 진실을 몰랐던 사람이라면
if (!truth[i]) {
truth[i] = true;
queue.add(i);
}
}
}
// 지민이가 거짓말을 할 수 있는 파티 개수
int answer = M;
for (int i = 0; i < M; i++) {
for (int j = 0, len = party[i].length; j < len; j++) {
// 진실을 알거나 들은 사람이 파티에 참석한다면
if (truth[party[i][j]]) {
answer--; break;
}
}
}
System.out.println(answer);
}느낀 점
처음엔 단순히 진실을 아는 사람을 만나고 난 이후에만 체크해주면 된다고 생각해서 파티에 오는 사람들을 입력 받으면서 진실을 아는 사람이 나오면 해당 파티만 다시 돌면서 진실을 알게된 여부를 체크해줬는데 문제를 다시 읽어보니 그게 아니었다...ㅎ 한 사람은 일관된 내용을 들어야 한다는 것! 그걸 빼먹었음😅
아무튼 반례 테스트 해보고 문제점을 찾아서 그걸 해결하기 위해 아이디어를 생각해봤는데 나한테는 인접 관계와 큐를 이용하는 방법이 가장 이해하기 쉽고 간단해서 이렇게 풀었다. 개인적으로 쉽지는 않았지만 풀면서 그래도 나름 재미있었다ㅋㅋㅋ 다음엔 문제 잘 읽고 더 빨리 풀 수 있기를✊
