[백준] #4358 생태학 (python)

4358번: 생태학

4358번: 생태학

문제

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

입력

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

출력

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

예제 입력 1

Red Alder
Ash
Aspen
Basswood
Ash
Beech
Yellow Birch
Ash
Cherry
Cottonwood
Ash
Cypress
Red Elm
Gum
Hackberry
White Oak
Hickory
Pecan
Hard Maple
White Oak
Soft Maple
Red Oak
Red Oak
White Oak
Poplan
Sassafras
Sycamore
Black Walnut
Willow

예제 출력 1

Ash 13.7931
Aspen 3.4483
Basswood 3.4483
Beech 3.4483
Black Walnut 3.4483
Cherry 3.4483
Cottonwood 3.4483
Cypress 3.4483
Gum 3.4483
Hackberry 3.4483
Hard Maple 3.4483
Hickory 3.4483
Pecan 3.4483
Poplan 3.4483
Red Alder 3.4483
Red Elm 3.4483
Red Oak 6.8966
Sassafras 3.4483
Soft Maple 3.4483
Sycamore 3.4483
White Oak 10.3448
Willow 3.4483
Yellow Birch 3.4483

✔️ Code

import sys
from collections import defaultdict
input = sys.stdin.readline

trees = defaultdict(int) # 기본 값이 0인 딕셔너리
tree_num = 0 # 나무의 총 개수

while True:
    tree = input().rstrip()
    if tree == '':
        break

    tree_num += 1
    trees[tree] += 1

trees_list = sorted(trees, key = lambda x: x[0])

for tree in trees_list:
    print('%s %.4f' %(tree, (trees[tree]/tree_num)*100))

✏️ Comment

딕셔너리로 구현해야겠다는 생각은 했는데, 미리 초기값을 다 0으로 받아놓고 싶은데 어떻게 해야할지 몰라서 구글링을 해본 결과, defaultdict(int)를 쓰면 해결할 수 있었다.

from collections import defaultdict
trees = defaultdict(int)

[파이썬 기초] 유사 딕셔너리 defaultdict() 활용법

[파이썬 기초] 유사 딕셔너리 defaultdict() 활용법

이 블로그의 글을 참고했다.

 

나는 사전 순으로 정렬해주기 위해서 sorted를 사용했는데 다른 블로그 보니까 아래처럼 한 사람도 많았다. 참고로 총 두 개의 자료형이 필요한데, tree의 이름을 출력해주기 위해서 list인 trees_list가 필요하고, 비율을 구하기 위해서 dictionary인 trees도 필요하다.

trees_list = list(keys())
trees_list.sort()

 

특히 출력 부분에서 이런 식으로 쓰는 방법을 외워둬야 할 것 같다. C언어와 비슷한 형태인 듯 하다. 여기서 %.4f 는 소수점 4번째 자리까지 출력하는 형식이다.

 print('%s %.4f' %(tree, (trees[tree]/tree_num)*100))