[Python Algorithm] 베스트셀러 BOJ #1302

문제

https://www.acmicpc.net/problem/1302

1302번: 베스트셀러

 

내 문제 풀이

import sys

n = int(sys.stdin.readline())
books = []
count = []
result = []

for _ in range(n):
    now = sys.stdin.readline().strip()
    if now in books:
        count[books.index(now)][1] += 1
    else:
        books.append(now)
        count.append([len(books)-1, 1])

count.sort(key = lambda x: x[1], reverse = True)

for i in range(len(count)):
    if count[i][1] == count[0][1]:
        result.append(books[count[i][0]])
    else:
        break

result.sort()
print(result[0])

👉 문자열 문제로, 팔린 책의 리스트와 책마다 몇 권 팔렸는지 저장하는 리스트를 따로 생성해서 풀었다.

• 입력받은 책이 books 리스트에 있는지 확인하고,
  없다면 books 리스트에 추가한 후, count에 [해당 책이 저장된 books 인덱스, 권 수]를 추가했다.
  있을 때는 books 리스트의 몇 번째 인덱스에 있는지 확인하여 count 리스트를 업데이트했다.
• 카운팅이 끝나면 count 리스트를 권 수를 기준으로 내림차순 정렬하고,
  제일 많이 팔린 책들을 result 리스트에 추가해주었다.
  그리고 result는 오름차순 정렬하여 맨 앞의 책만 출력하게 했다.

 

💡 피드백

• 어려운 문제는 아니었지만, 튜플이 수정 불가하다는 것을 모르고 있어서,, 그 부분에서 헷갈렸다.
  튜플은 수정 불가하므로, 리스트로 만들어서 연산을 해주어야 한다.
• 그리고 리스트에서 어떤 요소의 인덱스를 찾으려면 리스트명.index(요소) 이렇게 할 수 있다는 것도,,!