문제
지민이는 파티에 가서 이야기 하는 것을 좋아한다. 파티에 갈 때마다, 지민이는 지민이가 가장 좋아하는 이야기를 한다. 지민이는 그 이야기를 말할 때, 있는 그대로 진실로 말하거나 엄청나게 과장해서 말한다. 당연히 과장해서 이야기하는 것이 훨씬 더 재미있기 때문에, 되도록이면 과장해서 이야기하려고 한다. 하지만, 지민이는 거짓말쟁이로 알려지기는 싫어한다. 문제는 몇몇 사람들은 그 이야기의 진실을 안다는 것이다. 따라서 이런 사람들이 파티에 왔을 때는, 지민이는 진실을 이야기할 수 밖에 없다. 당연히, 어떤 사람이 어떤 파티에서는 진실을 듣고, 또다른 파티에서는 과장된 이야기를 들었을 때도 지민이는 거짓말쟁이로 알려지게 된다. 지민이는 이런 일을 모두 피해야 한다.
사람의 수 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
나의 풀이
- 이 문제는 교집합, 합집합 연산을 이용해 진실을 알 수 밖에 없는 파티원 집합을 구하고 이들이 참석하지 않는 파티의 수를 계산하는 문제이다. 처음에 이 문제를 읽은 후 진실을 아는 사람의 집합을 기준으로 파티를 순회하며 진실을 아는 사람 집합을 업데이트하면 될 것이라고 생각해 for문으로 1번 파티를 순회해 이 집합을 구하도록 구현했다. 이렇게 풀이했을 때 주어진 예제에 대해서는 모두 정답인데 제출만 하면 계속 틀려 대체 뭐가 잘못 된 건지 전혀 감이 안잡혔다. 그러나 새로운 예제를 기준으로 계산을 해보니 내가 구현한 방법대로면 파티 순서에 따라 진실을 아는 파티원 집합이 달라졌다... 아래 그림과 같이 입력이 주어질 때, 내가 구현한대로면 정답이 1인데 원래 정답은 0이어야 한다. 따라서 파티 순서에 관계없이 일정한 파티원 집합을 구하려면 중복 for문으로 파티를 순회해야 함을 깨닫고 아래와 같이 수정했다... 예외 케이스를 조금 더 꼼꼼하게 잡는 연습을 해야겠다...
import sys
input = sys.stdin.readline
answer = 0
N,M = map(int,input().split())
truths = list(map(int,input().split()))
parties = [list(map(int,input().split())) for _ in range(M)]
if len(truths)==1:
answer = M
else:
true_ones = set(truths[1:])
for i in range(M):
for party in parties:
party_ones = set(party[1:])
if(true_ones & party_ones):
true_ones = true_ones | party_ones
for i in range(M):
party_ones = set(parties[i][1:])
if not (true_ones & party_ones):
answer +=1
print(answer)
