[백준] 16165번: 걸그룹 마스터 준석이

www.acmicpc.net/problem/16165

16165번: 걸그룹 마스터 준석이

N, M = map(int, input().split())
dic = {}
for _ in range(N):
    group = str(input())
    memNum = int(input())
    memList = []
    for _ in range(memNum):
        memList.append(str(input()))
    memList.sort()
    dic[group] = memList

for _ in range(M):
    name = str(input())
    Q = int(input())
    if Q == 1:
        for k, v in zip(dic.keys(), dic.values()):
            if name in v:
                print(k)
                break             
    elif Q == 0:
        for mem in dic[name]:
            print(mem)

 

dict를 두 개 만드는 방법

N, M = map(int, input().split())
teamToMem = {}
memToTeam = {}

for _ in range(N):
    teamName = str(input())
    teamToMem[teamName] = []
    memNum = int(input())
    for _ in range(memNum):
        name = input()
        teamToMem[teamName].append(name)
        memToTeam[name] = teamName
    teamToMem[teamName].sort()

for _ in range(M):
    qName = str(input())
    qNum = int(input())
    if qNum: # 1이면 그룹명 출력
        print(memToTeam[qName])
    else:
        for mem in teamToMem[qName]:
            print(mem)