[Python] 백준 : 생태계 4358, 개수를 모르는 input, 소수점 4자리, EOFError 시간 초과, %

 

 

📝 문제

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

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

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

 

🫠 나의 풀이

# 시간 초과
while True:
    try:
        tree = input()
        total += 1

        if tree in dic:
            dic[tree] += 1
        else:
            dic[tree] = 1
    except EOFError:
        break

import sys
sys.stdin = open('input.txt', 'r')
# input = sys.stdin.readline

dic = {}
total = 0

while True:
    tree = input().strip()

    if tree == '':
        break

    total += 1

    if tree in dic:
        dic[tree] += 1
    else:
        dic[tree] = 1

sorted_dic = dict(sorted(dic.items()))

for key, value in sorted_dic.items():
    per = value / total * 100

    print("%s %.4f" % (key, per))

 

처음에 갯수를 모르는 input 을 받으니 EOFError 가 발생하면 while 문이 종료되도록 했는데 시간 초과가 떴다.

찾아보니 이는 종료 조건이 명시적이지 않고 예외 발생을 기다려야 해서 입력이 많을 경우에 느리고 비효율적이라고 한다.

 

➡️ if 문으로 명확한 조건을 설정해 빠르게 종료하자

 

🧞‍♂️ 다른 사람의 풀이

import sys

total = 0
dic = dict()
while 1:
    word = sys.stdin.readline().rstrip()
    if word == '':
        break
    total += 1   
    if word in dic:   # 전에 이미 나왔으면
        dic[word] += 1
    else:
        dic[word] = 1
sdic = dict(sorted(dic.items()))
for i in sdic:
    a = sdic[i]
    per = (a / total * 100)
    
    print("%s %.4f" %(i, per))

 

 

 

🔗 https://imzzan.tistory.com/29

[백준][Python] 4358번 생태학

🔗 https://blockdmask.tistory.com/534

[python] 파이썬 소수점 자리수 제한 방법 3가지

🔗 https://www.acmicpc.net/problem/4358