• TOC {:toc}

이 글은 백준 온라인 저지의 16165번 문제를 파이썬(Python)으로 풀이한 것을 모아놓은 글입니다.

일종의 연습 기록이며 제가 정답을 받은 코드와 참고할만한 다른 코드를 같이 기록합니다. 필요한 경우 코드에 대한 해설을 기록합니다만 코드는 통과했어도 해설은 틀릴 수 있기 때문에 가볍게 참고해주시길 부탁드립니다. 피드백은 편하신 방법으로 자유롭게 주시면 감사하겠습니다.

2021.08.09

메모리 시간 코드 길이
29200 KB 84 ms 466 B
단계 시작 시각 끝난 시각 걸린 시간
문제 이해 11:37:24 11:39:22
풀이 생각 11:39:28 11:40:07
코딩 11:40:08 11:50:15 
import sys
input = lambda: sys.stdin.readline().rstrip()
G = {}

N, M = map(int, input().split())
for _ in range(N):
    group = input()
    G[group] = []
    for _ in range(int(input())):
        member = input()
        G[member] = group
        G[group].append(member)
    G[group].sort()

for _ in range(M):
    q = input()
    is_member = int(input())
    if is_member:
        print(G[q])
    else:
        print("\n".join(G[q]))

아이디어 & 풀이

딕셔너리를 만들어서 그룹: [멤버]의 구조로 입력값을 저장한다.

멤버의 팀을 빠르게 찾기 위해 멤버: 팀도 저장한다.

  • 그룹: [멤버]만 저장했을 때는 다음과 같이 입력받은 멤버의 팀을 찾아 출력할 수 있다.
for group, members in G.items():
    if q in members:
        print(group)