[Baekjoon] 1043. 거짓말
문제 설명
문제
지민이는 파티에 가서 이야기 하는 것을 좋아한다. 파티에 갈 때마다, 지민이는 지민이가 가장 좋아하는 이야기를 한다. 지민이는 그 이야기를 말할 때, 있는 그대로 진실로 말하거나 엄청나게 과장해서 말한다. 당연히 과장해서 이야기하는 것이 훨씬 더 재미있기 때문에, 되도록이면 과장해서 이야기하려고 한다. 하지만, 지민이는 거짓말쟁이로 알려지기는 싫어한다. 문제는 몇몇 사람들은 그 이야기의 진실을 안다는 것이다. 따라서 이런 사람들이 파티에 왔을 때는, 지민이는 진실을 이야기할 수 밖에 없다. 당연히, 어떤 사람이 어떤 파티에서는 진실을 듣고, 또다른 파티에서는 과장된 이야기를 들었을 때도 지민이는 거짓말쟁이로 알려지게 된다. 지민이는 이런 일을 모두 피해야 한다.
사람의 수 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
예제 입력 2
4 1
1 1
4 1 2 3 4
예제 출력 2
0
예제 입력 3
4 1
0
4 1 2 3 4
예제 출력 3
1
예제 입력 4
4 5
1 1
1 1
1 2
1 3
1 4
2 4 1
예제 출력 4
2
예제 입력 5
10 9
4 1 2 3 4
2 1 5
2 2 6
1 7
1 8
2 7 8
1 9
1 10
2 3 10
1 4
예제 출력 5
4
예제 입력 6
8 5
3 1 2 7
2 3 4
1 5
2 5 6
2 6 8
1 8
예제 출력 6
5
예제 입력 7
3 4
1 3
1 1
1 2
2 1 2
3 1 2 3
예제 출력 7
0
출처
알고리즘 분류
문제 풀이
# Graph # Queue
Graph 또는 Queue를 이용하는 구현 문제입니다.
풀이 과정
문제 분류가 그래프이긴 한데, 저는 그래프 자료구조를 명시적으로 써서 풀지는 않았습니다. 어찌됐든 분리 집합이라는 개념은 이용하기 때문에, 그래프로 풀 수도 있을 것 같습니다.
제가 한 풀이는 아래와 같습니다.
-
진실을 알고 있는 사람이 많은 순으로 내림차순 정렬
-
앞에서부터 진실을 말하면서 새롭게 진실을 들은 사람들을 knowns에 추가
-
진실을 알고 있는 사람이 없을 때까지 1~2 반복
전체 코드
knowns를 new_knowns로 매번 갱신해주는 이유는 기존에 진실을 알고 있던 사람들이 parties에서 빠져나가면 남은 사람들 중에는 기존에 knowns에 있던 사람이 나올 가능성은 없기 때문에, 새롭게 진실을 알게 된 사람들인 new_knowns로 갱신해서 불필요한 후보들을 없앴습니다.
N, M = map(int,input().split())
knowns = set(list(map(int,input().split()))[1:])
parties = []
for _ in range(M):
party = list(map(int, input().split()))[1:]
parties.append(party+[sum([person in knowns for person in party])]) # [사람1, 사람2, ..., 진실을 알고 있는 사람수]
from collections import deque
while True:
parties = deque(sorted(list(parties),key=lambda x:x[-1], reverse=True))
if not parties or parties[0][-1] == 0:
break
new_knowns = set()
while parties and parties[0][-1] != 0:
party = parties.popleft()
new_knowns |= set([person for person in party[:-1] if person not in knowns])
knowns = new_knowns
for i in range(len(parties)):
parties[i][-1] = sum([person in knowns for person in parties[i]])
print(len(parties))
Leave a comment