문제집 추천, 이 문제집에 나온 유형들만 공부해도 코딩테스트는 거뜬: https://www.acmicpc.net/workbook/by/BaaaaaaaaaaarkingDog
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
예제 입력 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
이틀동안 끙끙댄 문제. 첫 시도는 모든 테스트 케이스에 맞는데 자꾸 10%에서 틀림이 떴다. 다른 모든 블로그는 다 똑같은 코드만 있고 다 마음에 들지 않았다. 그래프/BFS 문제에서 뭔가 아쉽다 싶으면 visited 리스트를 2개 만들자.
처음에는 remove를 이용하려 했었다.
초반에 제시한 내 코드
import sys
from collections import deque
if __name__=="__main__":
N,M=map(int, input().split()) # 사람의 수, 파티의 수
K=list(map(int, input().split()))
Party_list=[]
for _ in range(M):
info=list(map(int, input().split()))
Party_list.append(info[1:])
visited=[0]*(N+1)
if len(K)!=1:
Q=deque(K[1:])
while Q:
x=Q.popleft()
visited[x]=1
for party in Party_list:
if x in party:
Party_list.remove(party)
for y in party:
if visited[y]==0:
Q.append(y)
print(len(Party_list))
그래프/BFS 문제에서 remove를 쓰는거는 한번도 본적 없지만 위와 같이 풀었다. 이렇게 스스로 자신이 없으면 visited 리스트를 2개 만들자
정답
목표는 지민이를 전혀 모르는 사람들로 이루어진 파티수를 구하는 것이다.
즉, 지민이랑 만난적이 없으면서, 지민이와 만나사람을 만나지 않은 사람들로 이루어진 파티를 찾아야한다.
다시말해, 진실을 아는 사람과 진실을 아는 사람을 만난적 없는 사람들로 이루어진 파티 수
import sys
from collections import deque
if __name__=="__main__":
N,M=map(int, input().split()) # 사람의 수, 파티의 수
K=list(map(int, input().split()))
if len(K) == 1:
print(M)
exit()
Party_list=[]
for _ in range(M):
info=list(map(int, input().split()))
Party_list.append(info[1:])
visited_people=[0]*(N+1)
visited_party = [0] * M
Q=deque(K[1:]) # 진실을 아는 사람들
while Q:
x=Q.popleft() # 진실을 아는 사람 한명 빼서
visited_people[x]=1 # 확인한거 체크
for i in range(M): # 파티 수 별로 현재 사람을 체크한다
if x in Party_list[i]: # 진실을 아는 사람이 현재 파티에 속해 있으면
visited_party[i]=1 # 체크
for y in Party_list[i]: # 현재 파티에 참여한 사람들 에서
if visited_people[y]==0 and y!=x: # 이 파티에 진실을 아는 사람 제외, 처음 확인하는 사람
Q.append(y)
visited_people[y]=1 # 진실을 아는 사람이 속한 파티의 파티원들은 진실을 알게 됨
print(visited_party.count(0))
내 패러그래프 방향은 맞았는데 구현이 아쉬웠다.
