[ BOJ ] 4358 : 생태학 ( SILVER 2 ) / Python

>> 문제 바로가기 (https://www.acmicpc.net/problem/4358)

문제

생태학에서 나무의 분포도를 측정하는 것은 중요하다. 그러므로 당신은 미국 전역의 나무들이 주어졌을 때, 각 종이 전체에서 몇 %를 차지하는지 구하는 프로그램을 만들어야 한다.

입력

프로그램은 여러 줄로 이루어져 있으며, 한 줄에 하나의 나무 종 이름이 주어진다. 어떤 종 이름도 30글자를 넘지 않으며, 입력에는 최대 10,000개의 종이 주어지고 최대 1,000,000그루의 나무가 주어진다.

출력

주어진 각 종의 이름을 사전순으로 출력하고, 그 종이 차지하는 비율을 백분율로 소수점 4째자리까지 반올림해 함께 출력한다.

풀이 과정

특정 나무가 몇 번 등장했는지를 카운트하고. 전체 나무 개수에 대한 백분율을 계산한다.

종의 이름을 사전순대로 출력해야 하므로 정렬 과정이 필요하다.

입력의 끝 표시가 주어지지 않으므로 EOF에 대한 대비도 필요하다.

import sys
input = sys.stdin.readline

trees = {}
cnt = 0
while True:
    tree = input().rstrip()
    if tree == '': break
    if trees.get(tree, -1) == -1: trees[tree] = 1
    else: trees[tree] += 1
    cnt += 1

result = list(trees.items())
result.sort(key=lambda x:x[0])

for r in result:
    print(r[0], '%.4f'%(r[1]/cnt*100))