[BOJ][⚪2][백준#04358] 생태학

문제 출처

BAEKJOON Online Judge #4358

문제

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

입력

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

출력

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

예제

입력

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

출력

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

딕셔너리를 이용해 카운트를 하고, 이를 출력 형식에 맞춰서 출력하는 방식의 풀이였다. 종료코드 없이 무작정 입력을 마쳐버리는 스타일의 문제였다.

My Sol A: try -except -else 문

import sys
input = sys.stdin.readline

dic = dict()
cnt = 0
while True:
    try:
        a = input()
    except:
        break
    else:
        dic[a] = dic.get(a, 0) + 1
        cnt += 1

for tree in sorted(dic.keys()):
    print(f'{tree} {dic[tree]*100/cnt:.4f}')

처음에는 try -except -else 문을 이용해 입력을 받으려고 try했을 때 EOF 에러가 발생한다면 break, 아니면 이후 카운트를 진행하는 방식으로 풀이를 짰는데, 이것이 매번 if문을 반복하기 때문에 시간 초과가 발생한 것 같다.

My Sol B:

import sys

dic = dict()
cnt = 0
for a in sys.stdin:
    a = a.rstrip()
    dic[a] = dic.get(a, 0) + 1
    cnt += 1

for tree in sorted(dic.keys()):
    print(f'{tree} {dic[tree]*100/cnt:.4f}')

그래서 전체 입력을 sys.stdin으로 받아버리고, 이 라인들을 처리해서 출력하는 방식을 사용하였다. 이 방법을 사용한다면 자연스럽게 끝라인까지 iterable을 실행할 수 있다.

이때 주의할 점은, 입력이 모두 \n이 붙어있어 rstrip() 해주어야 형식에 맞게 출력된다는 것이다. 그리고 한편으로는, .4f 등의 출력 형식에 대한 이해가 있어야 쉽게 풀 수 있는 문제였다.

결과

맞았습니다!!