[🥈4 / 백준 1302 / 파이썬] 베스트셀러

 

1302번: 베스트셀러

 

문제

김형택은 탑문고의 직원이다. 김형택은 계산대에서 계산을 하는 직원이다. 김형택은 그날 근무가 끝난 후에, 오늘 판매한 책의 제목을 보면서 가장 많이 팔린 책의 제목을 칠판에 써놓는 일도 같이 하고 있다.

오늘 하루 동안 팔린 책의 제목이 입력으로 들어왔을 때, 가장 많이 팔린 책의 제목을 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 오늘 하루 동안 팔린 책의 개수 N이 주어진다. 이 값은 1,000보다 작거나 같은 자연수이다. 둘째부터 N개의 줄에 책의 제목이 입력으로 들어온다. 책의 제목의 길이는 50보다 작거나 같고, 알파벳 소문자로만 이루어져 있다.

출력

첫째 줄에 가장 많이 팔린 책의 제목을 출력한다. 만약 가장 많이 팔린 책이 여러 개일 경우에는 사전 순으로 가장 앞서는 제목을 출력한다.

 

예제 입력 1 5 top top top top kimtop	예제 출력 1  top
예제 입력 2 9 table chair table table lamp door lamp table chair	예제 출력 2 table
예제 입력 3 6 a a a b b b	예제 출력 3 a
예제 입력 4 8 icecream peanuts peanuts chocolate candy chocolate icecream apple	예제 출력 4 chocolate
예제 입력 5 1 soul	예제 출력 5 soul

 

풀이

 

 

각 문자열을 입력받은 횟수를 구한 뒤 최고 순위를 출력해야 한다.

 

문자열을 입력받을 때, 기존에 있던 문자열인지 확인하기 위해 탐색을 진행해야 한다.

여기서 리스트로 진행한다면 탐색에 O(n)의 시간이 소요되므로 총 O(n^2)의 알고리즘이 될 것이다.

따라서 탐색 속도가 O(1)인 해시 맵, 즉 딕셔너리를 이용하여 문제를 해결해야 한다.

 

모든 문자열을 받았다면 딕셔너리에는 각 문자열이 책 이름 : 횟수, 즉 key : value 로 매칭이 되어 있을 것이다.

우린 이 딕셔너리의 원소들을 받아 value가 가장 높은 key를 출력해야 하므로 정렬을 활용한다.

 

여기서 조건이 하나 더 있는데, 사전 순으로 가장 앞서는 제목을 출력해야 한다는 것이다.

따라서 value 를 내림차순으로, key를 오름차순으로 정렬하여 맨 앞에 있는 것을 출력한다.

 

import sys
input = sys.stdin.readline

d = dict()
n = int(input())
for _ in range(n):
    book = input().rstrip()
    # 이미 입력받았던 책인지 확인
    if d.get(book):
        d[book] += 1
    else:
        d[book] = 1
# 람다식을 활용하여 정렬
print(sorted(list(d.items()), key = lambda x : (-x[1],x[0]))[0][0])