41-H0ngJu

[H0ngJu]

🔗 문제 링크

루머

✔️ 소요된 시간

1.5H?

✨ 수도 코드

와 이 문제 잘못 읽어가지고 몇번을 다시 풀었는지 모르겠습니다 ... ㅋㅋ

매분 루머를 믿는 사람은 모든 주변인에게 루머를 동시에 퍼트리며, 군중 속 사람은 주변인의 절반 이상이 루머를 믿을 때 본인도 루머를 믿는다.

이 부분을 주의해서 해석해야 합니다..

7
2 4 0
1 3 0
2 5 0
1 5 6 0
3 4 6 7 0
4 5 7 0
5 6 0
1
6

처럼 데이터가 들어오면, 6번이 최초유포자입니다.

그러면 6번의 주변인인 4, 5, 7이 동시에 루머를 믿는 것이 아니라

4의 주변인의 절반이 루머를 믿을 때,
5의 주변인의 절반이 루머를 믿을 때,
7번의 주변인의 절반이 루머를 믿을 때,

루머를 믿게 되는 것입니다.

즉 time = 1일 때,
4의 주변인 3명중 6만 루머를 믿으므로 4는 믿지 않습니다.
5의 주변인 4명중에 6만 루머를 믿으므로 5는 믿지 않습니다.
7의 주변인 2명중에 6이 루머를 믿으므로 (절반이 믿음) 7은 루머를 믿게 됩니다.

time = 2일 때,
7의 주변인 중에는 5,6을 보면,
5의 주변인 4명중에 절반인 2명이 (6,7)이 믿으므로 5도 루머를 믿기 시작합니다.

요런 메커니즘인거죠....

위처럼 시간의 순서대로 퍼져나가는 걸보면,
deque를 통해서 BFS로 미로 찾듯이 하면 풀면 되겠다! 하는 가닥이 보입니다.

위 흐름대로 고~~대로 작성하면,

while q:
    person, time = q.popleft()
    
    cnt = 0
    for p in info[person]:
        if p == 0:
            break
        if answer[p] == -1:
            belive[p] += 1
            
            if belive[p] >= (len(info[p]) + 1) // 2:
                answer[p] = time + 1
                q.append((p, time+1)) 

저는 각 사람마다 주변인들 중 루머를 믿는 사람의 수를 갱신하도록 구현했습니다.

믿는 사람을 list, set으로 관리를 했는데, 명수로 관리하는게 가장 깔끔한 방법인듯 합니다.

끗!

📚 새롭게 알게된 내용

[tgyuuAn]

import sys
from collections import defaultdict

def input(): return sys.stdin.readline().rstrip()

N = int(input())
link = defaultdict(list)
now = defaultdict(set)

for idx in range(1,N+1):
    link_info = list(map(int, input().split()))
    link_info.pop()
    link[idx] = link_info
    now[idx] = set(link_info)

remain_count = int(input())
remain = list(map(int, input().split()))

answer = [-1 for _ in range(N+1)]
visited = set()
turn = 0
while remain:
    new = list()
    
    for lier in remain:
        if lier in visited: continue
        answer[lier] = turn
        visited.add(lier)
        
        for neighbor in link[lier]:
            if neighbor in visited: continue
        
            now[neighbor].discard(lier)
            if len(link[neighbor]) >= len(now[neighbor])*2:
                new.append(neighbor)
                            
    remain = new
    turn += 1
    
print(*answer[1:])

헉 나도 List, set으로 했는데...!

set이 편하다고요~~~