우선순위 큐(유형정리) - 백준 4358

4358번: 생태학 (acmicpc.net)

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

 

딱 보면 아! Map을 쓰는 건가? 라는 생각이 드는 문제였다. Python에서는 딕셔너리라는 문법으로 Map을 구현하고 있기 때문에 dictionary를 사용해서 문제를 해결했다.

 

※ 알고리즘

1. 반복적으로 나무의 이름을 입력받는다.

2. 이름이 없을 경우 break 시켜준다.

3. dictionary 내부에 이름이 있을 경우, 전에 이름이 출력 되었으면 +1 시켜주고, 전에 이름이 출력된 적이 없을 경우 처음 값이므로 1을 대입해준다.

4. 최종 count를 해서 평균을 구해야하므로 변수 갯수를 마지막에 카운팅 해준다.

5. dictionary의 key를 리스트로 만들고, 정렬 시켜준다.

6. dictionary의 key 리스트 내부에 나무의 이름이 있으면 반복적으로 출력 형식에 맞추어 출력시켜준다.

 

import sys
input = sys.stdin.readline

tree_kind = {}

count = 0
while True:
    name = input().rstrip()
    if not name:
        break
    if name in tree_kind:
        tree_kind[name] += 1     #전에 나온 경우 +1
    else:
        tree_kind[name] = 1      #처음 나온 경우 : 1대입

    count += 1      #변수 갯수 카운팅

tree = list(tree_kind.keys())
tree.sort()

for name in tree:
    print('%s %.4f' %(name, tree_kind[name]/count * 100))

 

이렇게 쿤제를 해결할 수 있었다. 이 문제에서 처음엔 시간초과가 나서 sys.stdin.readline을 포함해주었는데 이제는 출력형식이 맞지 않다는 문구를 만났다. 이게 뭔 일인고... 해서 구글링을 해보니, sys.stdin.readline은 개행문자 '\n'을 자동으로 포함한다고 한다. 그래서 오른쪽을 제거하는 rstrip()을 포함시켜주어야 올바른 출력 형식으로 정답이 나오게 되었다.

 

좀 어이없는 출력 형식 억까로 인해, 꽤 오랜시간을 붙잡고 있던 문제였다... ㅋㅋㅋ