[백준] Python - 20291 파일 정리

문제 링크

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

20291번: 파일 정리

 

문제 풀이

1. 입력받은 문자열에서 확장자만 잘라낸다. 
2. 확장자를 키값으로 가지는 딕셔너리에 값을 업데이트 한다.
   1. 만약 이미 딕셔너리에 추가된 확장자라면 값을 1 더해준다.
   2. 딕셔너리에 없는 확장자라면, 새롭게 확장자를 키값으로 값을 1 추가해준다.
3. 딕셔너리의 키값을 기준으로 오름차순 정렬하여, 키와 값을 출력한다.

 

다음엔 2번의 복잡한 과정 대신 defaultdict를 사용해 봐야겠다!

 

전체 코드

from sys import stdin
input = stdin.readline

num = int(input())
ext = {}

for _ in range(num) :
    file = input()
    index = file.find(".")
    e = file[index+1:].strip()

    if e in ext :
        ext[e] += 1
    else : ext[e] = 1

for (key, value) in sorted(ext.items()) :
    print(key, value)