[백준/Python] BOJ 16165 - 걸그룹 마스터 준석이

👀 문제

16165번: 걸그룹 마스터 준석이 ✨ 실버 3

정우는 소문난 걸그룹 덕후이다. 정우의 친구 준석이도 걸그룹을 좋아하지만 이름을 잘 외우지 못한다는 문제가 있었다.

정우는 친구를 위해 걸그룹 개인과 팀의 이름을 검색하여 외우게 하는 퀴즈 프로그램을 만들고자 한다.

🔑 키포인트

자료 구조

✍️ 코드

import sys

# 총 입력받을 걸그룹의 수 n
# 맞혀야 할 문제의 수 m
n, m = map(int, sys.stdin.readline().split())
group = "" # 그룹명
num = 0 # 멤버 수

groups = {} # 그룹 담을 dictionary

for _ in range(n):
    members = []
    group = sys.stdin.readline().rstrip()
    num = int(sys.stdin.readline())
    for _ in range(num):
        members.append(sys.stdin.readline().rstrip())
    # 그룹 딕셔너리에 그룹이름 : 멤버들 추가
    groups[group] = members

question = {} # 문제 딕셔너리

for _ in range(m):
    fir = sys.stdin.readline().rstrip()
    sec = int(sys.stdin.readline())
    question[fir] = sec

for q in list(question.keys()):
    if question[q] == 1:
        for k, v in groups.items():
            if q in v:
                print(k)
    else:
        # 멤버 이름 순서대로 출력
        for i in sorted(list(groups[q])):
            print(i)