[1043] 거짓말 - Python

/* 문제 */
지민이는 파티에 가서 이야기 하는 것을 좋아한다. 파티에 갈 때마다, 지민이는 지민이가 가장 좋아하는 이야기를 한다. 지민이는 그 이야기를 말할 때, 있는 그대로 진실로 말하거나 엄청나게 과장해서 말한다. 당연히 과장해서 이야기하는 것이 훨씬 더 재미있기 때문에, 되도록이면 과장해서 이야기하려고 한다. 하지만, 지민이는 거짓말쟁이로 알려지기는 싫어한다. 문제는 몇몇 사람들은 그 이야기의 진실을 안다는 것이다. 따라서 이런 사람들이 파티에 왔을 때는, 지민이는 진실을 이야기할 수 밖에 없다. 당연히, 어떤 사람이 어떤 파티에서는 진실을 듣고, 또다른 파티에서는 과장된 이야기를 들었을 때도 지민이는 거짓말쟁이로 알려지게 된다. 지민이는 이런 일을 모두 피해야 한다.
사람의 수 N이 주어진다. 그리고 그 이야기의 진실을 아는 사람이 주어진다. 그리고 각 파티에 오는 사람들의 번호가 주어진다. 지민이는 모든 파티에 참가해야 한다. 이때, 지민이가 거짓말쟁이로 알려지지 않으면서, 과장된 이야기를 할 수 있는 파티 개수의 최댓값을 구하는 프로그램을 작성하시오.

이 문제는 작년에 그래프를 모르던 상태에서 풀려고 시도하다가 실패했던 문제다.

 

dfs로 구현했다.

# 정답 코드
import sys
sys.setrecursionlimit(10**9)

def dfs(graph, v, heard):
    if heard[v] == 1:   # 진실을 아는 사람일 경우
        for i in graph[v]:
            if heard[i] == 0:
                heard[i] = 1
                dfs(graph, i, heard)

n, m = map(int, input().split())
t = list(map(int,input().split()))    # 진실을 아는 사람들
del t[0]

# 아무 것도 못들은 사람: 0, 진실을 아는 사람: 1
heard = [0 for _ in range(n+1)]
for i in t:
    heard[i] = 1

# 같은 파티에 오는 사람들끼리 연결
graph = [[] for _ in range(n+1)]
parties = []
for _ in range(m):
    party = list(map(int, input().split()))
    del party[0]
    parties.append(party)
    for a in party:
        for b in party:
            if a != b:
                graph[a].append(b)

for i in t:
    dfs(graph, i, heard) # 진실을 아는 사람들에 대해 dfs

res = 0
for party in parties:
    possible = True
    for p in party:
        if heard[p] == 1:
            possible = False
            break
    if possible:
        res += 1

print(res)

각 사람에는 번호가 주어진다. heard 리스트에 아무 것도 못들은 사람은 0, 진실을 아는 사람은 1로 표시하고 graph 리스트에 같은 파티에 오는 사람들을 각각 저장한다. 그리고 진실을 아는 사람 모두를 각각 시작점으로 dfs를 수행하면 진실을 말해야만 하는 사람들도 모두 1로 표시된다. 파티에 진실을 아는 사람이 한 명이라도 있으면 그 파티에서는 거짓을 말할 수 없다. 따라서 진실을 아는 사람이 한 명도 없는(참여자의 heard의 값이 모두 0인) 파티의 수를 출력한다.