[BaekJoon] 4358번 : 생태학 (Python)

코딩 1일 1문제! 오늘의 문제는 백준의 생태학 입니다.

백준의 생태학을 파이썬으로 풀이하였습니다.

4358번: 생태학

👨🏻‍💻 문제 풀이

생태학 이 문제는.... 여러번의 시간 초과 이후에 백준의 흔한 시간 초과 해결 방법인

input() -> sys.stdin.readline() 변경 이후 맞았습니다를 볼 수 있었던 문제입니다...

while 반복문을 잘 못 다뤄서 시간 초과가 나는건가 생각되어 한참 고민했던 문제입니다.

 

이 문제는 Dictionary를 잘 활용하면 풀 수 있는 문제입니다.

Dictionary에 나무이름을 키로하고 각 나무의 개수를 해당 나무 이름 키의 위치에서 증가 시켜준 뒤

tree_dict = {}
    tree_name_list = set()
    
    tree_num = 0
    
    while True:
        tree = input().rstrip()
        
        if not tree:
            break
        tree_num += 1
        
        if tree not in tree_dict.keys():
            tree_dict[tree] = 1
            tree_name_list.add(tree)
        else:
            tree_dict[tree] += 1
    
    tree_names = sorted(list(tree_name_list))

 

나중에 하나씩 꺼내어 전체 나무개수로 나누고 100을 곱하여 백분율을 구해주고

이를 f-string을 활용하여 원하는 모양으로 출력하면서 소수점 4자리 수 까지 .4f 로 표현합니다.

tree_dict, tree_names, tree_num = ecology()
    
    for tree in tree_names:
        print(f"{tree} {((tree_dict[tree] / tree_num) * 100):.4f}")

그런데 여기서 input() 을 그대로 사용하면! 시간초과... 가 발생합니다.

from sys import stdin
input = stdin.readline

이에 맨 위에 input을 sys.stdin.readline으로 바꾸어주는 코드를 넣어주면 맞았습니다! 를 보실 수 있습니다.

 

전체 코드는 아래를 참고해주세요.

👨🏻‍💻 코드 ( Solution )

from sys import stdin
input = stdin.readline

def ecology():
    tree_dict = {}
    tree_name_list = set()
    
    tree_num = 0
    
    while True:
        tree = input().rstrip()
        
        if not tree:
            break
        tree_num += 1
        
        if tree not in tree_dict.keys():
            tree_dict[tree] = 1
            tree_name_list.add(tree)
        else:
            tree_dict[tree] += 1
    
    tree_names = sorted(list(tree_name_list))
    
    return tree_dict, tree_names, tree_num


if __name__ == "__main__":
    tree_dict, tree_names, tree_num = ecology()
    
    for tree in tree_names:
        print(f"{tree} {((tree_dict[tree] / tree_num) * 100):.4f}")

GitHub - SOMJANG/CODINGTEST_PRACTICE: 1일 1문제 since 2020.02.07