[algorithm] 백준 20291 - 파일 정리 (파이썬)

📚 문제

입력

출력

입력 예시

8
sbrus.txt
spc.spc
acm.icpc
korea.icpc
sample.txt
hello.world
sogang.spc
example.txt

출력 예시

icpc 2
spc 2
txt 3
world 1

🧑🏻‍💻 풀이 과정

• n만큼 반복하면서, 입력받은 문자에서 확장자명만 저장하자.
• 확장자명이 딕셔너리에 없으면 key=확장자명, value=0으로 초기화하고 +1, 있으면 그냥 +1
• 확장자명을 오름차순 정렬 후, 출력하자.

def file_clean(n):
    ans = {}
    for _ in range(n):

        # 입력받은 문자에서 확장자만 저장하자.
        split_s = input().split(".")[1]

        # 확장자가 없으면 0으로 초기화한 후 1 더하고, 있으면 그냥 1 더하자.
        ans.setdefault(split_s, 0)
        ans[split_s] += 1

    # 확장자 오름차순 정렬
    for k, v in sorted(ans.items(), key=lambda x: x[0]):
        print(k, v)


n = int(input())
file_clean(n)