99클럽 코테 스터디 32일차 TIL + 해시

문제

백준 걸그룹 마스터 준석이 https://www.acmicpc.net/problem/16165

 

코드

n, m = map(int, input().split())    # 걸그룹의 수, 맞혀야 할 문제의 수

idol_dic = {}
for _ in range(n):
    team_name = input()
    team_member_num = int(input())
    team_member = []
    for _ in range(team_member_num):
        team_member.append(input())

    idol_dic[team_name] = team_member


for _ in range(m):
    quiz = input()
    quiz_type = int(input())

    if quiz_type == 0:  # 팀 이름 주어짐
        for member in sorted(idol_dic[quiz]):
            print(member)

    else:   # 팀 멤버 이름 주어짐
        for team, member in idol_dic.items():
            if quiz in member:
                print(team)
                break

 

풀이

n, m = map(int, input().split())    # 걸그룹의 수, 맞혀야 할 문제의 수

입력 형식에 맞춰 걸그룹의 수 n과 맞춰야할 문제의 수 m을 입력받는다.

 

idol_dic = {}
for _ in range(n):
    team_name = input()
    team_member_num = int(input())
    team_member = []
    for _ in range(team_member_num):
        team_member.append(input())

    idol_dic[team_name] = team_member

idol_dic 딕셔너리를 선언한다. idol_dic에는 {팀 이름 : 팀 멤버들}의 정보를 담는다.

 

걸그룹의 수 만큼 {팀 이름 : 팀 멤버들}의 정보를 입력받는다.

팀 이름(team_name)과 팀 인원수(team_member_num)를 입력받는다.

팀 멤버의 이름을 담을 배열 team_member를 만들고 멤버 이름을 하나하나 입력 받아 배열에 차곡차곡 추가한다.

 

for _ in range(m):
    quiz = input()
    quiz_type = int(input())

그 다음, m개의 퀴즈를 입력받는다.

quiz 변수에 팀 이름이나 멤버의 이름을 받고, 

quiz_type에 0 또는 1을 입력받는다.

 

if quiz_type == 0:  # 팀 이름 주어짐
    for member in sorted(idol_dic[quiz]):
        print(member)

else:   # 팀 멤버 이름 주어짐
    for team, member in idol_dic.items():
        if quiz in member:
            print(team)
            break

quiz_type이 0인 경우 즉, 팀의 이름이 주어진 경우는

팀에 속한 멤버의 이름들을 사전순으로 정렬해서 출력한다.

 

quiz_type이 1인 경우 즉, 팀 멤버의 이름이 주어진 경우는

해당 멤버가 속한 팀 이름을 출력한다.

 

다른 코드

n, m = map(int, input().split())    # 걸그룹의 수, 맞혀야 할 문제의 수
idol_dic = {}

for _ in range(n):
    team_name = input()
    team_member_num = int(input())
    team_member = []
    for _ in range(team_member_num):
        team_member.append(input())

    idol_dic[team_name] = team_member


for _ in range(m):
    quiz = input()
    quiz_type = int(input())

    if quiz_type == 0:  # 팀 이름 주어짐
        for member in sorted(idol_dic[quiz]):
            print(member)
        break

    else:   # 팀 멤버 주어짐
        for team in idol_dic.keys():
            if quiz in idol_dic[team]:
                print(team)
                break

 

풀이

n, m = map(int, input().split())    # 걸그룹의 수, 맞혀야 할 문제의 수
idol_dic = {}

for _ in range(n):
    team_name = input()
    team_member_num = int(input())
    team_member = []
    for _ in range(team_member_num):
        team_member.append(input())

    idol_dic[team_name] = team_member


for _ in range(m):
    quiz = input()
    quiz_type = int(input())

    if quiz_type == 0:  # 팀 이름 주어짐
        for member in sorted(idol_dic[quiz]):
            print(member)
        break

전 코드와 동일하다.

 

else:   # 팀 멤버 주어짐
    for team in idol_dic.keys():
        if quiz in idol_dic[team]:
            print(team)
            break

전 코드에서는 딕셔너리의 item() 함수를 사용했다.

이 코드에서는 .keys() 함수를 이용해 딕셔너리의 키값들만 따로 모아서

해당 키값의 밸류들(=멤버들) 사이에 퀴즈로 나온 멤버가 있다면, 

그 멤버의 팀 이름을 출력한다.

 

새롭게 배운점 / 느낀점

• 딕셔너리 이름이 book이라고 했을때,
  • book.keys() : book의 키들만 모아 dict_keys 객체를 반환한다.
  • book.items() : book에 있는 키와 밸류들의 쌍을 얻을 수 있다.
    • ex ) book = {'교과서' : '국어', '과학'} ➡️ dict_items([('교과서' : '국어'), ('교과서' : '과학')])

• 리스트만 써오다가 딕셔너리를 활용하려니 능숙하게 다루기가 아직 어렵다. 딕셔너리로 풀 수 있는 문제는 적극적으로 딕셔너리로 풀며 숙련도를 높여가야겠다.

 

다른 의견이나 궁금하신게 있다면 편하게 댓글 달아주세요 :)