[백준] 16165번 걸그룹 마스터 준석이 파이썬 코드

코드

import sys
input = sys.stdin.readline
n,m = map(int, input().split())

girlgroup = {} # 1번 유형 퀴즈 답변을 위한 dict
girlgroupname = {} # 0번 유형 퀴즈 답변을 위한 dict

for _ in range(n):
  x = input().rstrip()
  member = int(input())
  for _ in range(member):
    name = input().rstrip()
    girlgroup[name] = x
    # 이 부분은 for문 처음에 리스트를 만들어 마지막에 넣는 방법으로 최적화할 수 있는 듯하다.
    if x not in girlgroupname:
      girlgroupname[x] = [name]
    else: girlgroupname[x].append(name)

for _ in range(m):
  quiz = input().rstrip()
  categorynum = int(input())
  if categorynum == 1:
    print(girlgroup[quiz])
  elif categorynum == 0:
    print(*sorted(girlgroupname[quiz]), sep='\n')

dict를 2개 만들어서 해결할 수 있었던 문제였다.