📝 문제
💡 정우는 소문난 걸그룹 덕후이다.
정우의 친구 준석이도 걸그룹을 좋아하지만 이름을 잘 외우지 못한다는 문제가 있었다.
정우는 친구를 위해 걸그룹 개인과 팀의 이름을 검색하여 외우게 하는 퀴즈 프로그램을 만들고자 한다.
⚙️ 입력 : 첫 번째 줄에는 총 입력 받을 걸그룹의 수 N과 맞혀야 할 문제의 수 M을 입력받는다.
두 번째 줄부터는 각 걸그룹마다
팀의 이름, 걸그룹의 인원 수, 멤버의 이름을 한 줄씩 차례대로 입력받는다.
팀과 멤버의 이름은 최대 100글자이며, 모든 글자는 알파벳 소문자이다.
하나의 걸그룹이나 서로 다른 두 걸그룹에 이름이 같은 두 멤버가 있는 경우는 없다.
그 다음 줄부터는 M개의 퀴즈를 입력받는다.
각각의 퀴즈는 두 줄로 이루어져 있으며,
팀의 이름이나 멤버의 이름이 첫 줄에 주어지고 퀴즈의 종류를 나타내는 0 또는 1이 두 번째 줄에 주어진다.
퀴즈의 종류가 0일 경우 팀의 이름이 주어지며, 1일 경우 멤버의 이름이 주어진다.
⚙️ 출력 : 첫 번째 줄부터 차례대로 퀴즈에 대한 답을 출력한다.
퀴즈의 종류가 0일 경우 해당 팀에 속한 멤버의 이름을 사전순으로 한 줄에 한 명씩 출력한다.
퀴즈의 종류가 1일 경우 해당 멤버가 속한 팀의 이름을 출력한다.
16165번: 걸그룹 마스터 준석이
정우는 소문난 걸그룹 덕후이다. 정우의 친구 준석이도 걸그룹을 좋아하지만 이름을 잘 외우지 못한다는 문제가 있었다. 정우는 친구를 위해 걸그룹 개인과 팀의 이름을 검색하여 외우게 하는
www.acmicpc.net
✅ 풀이
n, m = map(int, input().split())
otaku = {}
for i in range(n):
group = input()
members = []
for j in range(int(input())):
members.append(input())
otaku[group] = members
for i in range(m):
name = input()
type = int(input())
if type == 0:
for item in sorted(otaku[name]):
print(item)
else:
print([k for k, v in otaku.items() if name in v][0])💡 otaku라는 딕셔너리에 그룹명을 키로, 멤버 리스트를 값으로 저장했다.
이후 문제 타입이 0이면, 해당 그룹 멤버들을 sorted로 정렬하여 출력했고,
1이면 해당 멤버가 속하는 그룹의 키를 찾아서 출력했다.
