BOJ : 20291 파일정리 (파이썬)

문제

https://www.acmicpc.net/problem/20291

20291번: 파일 정리

풀이

우선 문제를 보면 파일명을 기준으로 정렬을 실행해야 할거같은 생각이 들었다. 그래서 일단 split()을 통해 ' . ' 을 구분해주고 확장자만 뽑아서 딕셔너리에 key value를 통해 어떤 확장자가 몇번 나왔는지 카운트를 해줬다.

소스코드

import sys

n = int(sys.stdin.readline())

dict = {}

for _ in range(n) :
    s = sys.stdin.readline().split('.')
    s[1] = s[1].rstrip()
    if s[1] not in dict :
        dict[s[1]] = 1 
    else :
        dict[s[1]] += 1 

sort_dict = sorted(dict.keys())

for i in range(len(dict)) :
    print(sort_dict[i],end=' ')
    print(dict[sort_dict[i]])