[백준] 1043. 거짓말 : python

https://www.acmicpc.net/problem/1043

1043번: 거짓말

문제

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

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

입력

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

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

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

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

출력

첫째 줄에 문제의 정답을 출력한다.

---

풀이

무려 4중 for문을 사용해서 문제를 해결했다. 입력 값의 최대 값이 모두 50이하라서 통과지 조금만 더 컸다면 시간초과로 통과하지 못했을 효율적이지 않은 코드라고 생각한다.

 

일단 풀이는

진실을 알고 있는 사람을 list에 담아준다.

 

그리고 파티에 방문하여 그 파티에 사람들 중 진실을 알고 있는 사람이 있는지 검사한다.

만약 진실을 알고 있는 사람이 있다면 그 파티에 있는 모든 사람들을 진실을 알고 있는 리스트에 추가한다

해당 파티를 cnt 에 저장한다.

모든 파티를 돌고 나서 모든 파티의 개수(M) - 진실을 아는 사람이 있는 파티(cnt)의 갯수를 뺀다

 

이렇게 하면 문제가 발생한다!

예제 5번의 경우에 7번째 파티에 10번은 진실을 모르는 상태이기 때문에 지민이는 10번에게 과장된 말을 하고

이후에 8번째 파티에서 과장된 진실을 아는 10번과 진실을 아는 3번이 만나게 되는 것이다...!＼(º □ º l|l)/

그럼 꼼짝 없이 지민이는 거짓말쟁이가 되버린다.

 

이걸 해결하기 위해 해당 과정을 파티의 개수만큼 반복해줬다

그럼 cnt의 값이 어마무시하게 많아지는데 set()으로 중복을 제거하면 해결!

 

다른 사람들의 코드를 보고 더 효율적인 방법을 찾아야겠다.

python 코드

# 1043 거짓말
N, M = map(int,input().split())
know = list(map(int,input().split()))
knowNum = know[0]
know.pop(0)
party = []
cnt = []

for i in range(M):
    p = list(map(int,input().split()))
    p.pop(0)
    party.append(p)
for a in range(M):
    for i in range(M):
        for j in range(len(know)):
            if know[j] in party[i]:
                cnt.append(i)
                for k in party[i]:
                    if k not in know:
                        know.append(k)
                break
cnt = set(cnt)
print(M - len(cnt))