[백준] 20291번 파일 정리

💡문제접근

• 파일명은 필요하지 않으므로 .을 기준으로 뒤에 있는 확장자를 찾아서 딕셔너리에 저장한 후 딕셔너리를 확장자 이름을 기준으로 오름차순 정렬한다.

💡코드(메모리 : 43708KB, 시간 : 220ms)

import sys
N = int(input())

file = {}
for _ in range(N):
    file_name = sys.stdin.readline().strip()
    file_name = file_name[file_name.index(".") + 1:]
    if file_name in file:
        file[file_name] += 1
    else:
        file[file_name] = 1

file = sorted(file.items(), key = lambda x : (x[0]))
for i in file:
    print(i[0], i[1])

📌 PyPy3로 제출한 결과 메모리 : 133952KB, 시간 : 388ms

💡소요시간 : 2m