[백준 20291번] 파일 정리

20291번: 파일 정리

import sys
from collections import Counter
input = sys.stdin.readline

n = int(input().rstrip())

texts = [input().rstrip().split(".")[1] for _ in range(n)]

for name, count in sorted(list(dict(Counter(texts)).items()),key=lambda x: x[0]):
    print(f"{name} {count}")

 

이번 문제는 확장자 별로 파일의 개수를 구하여 출력하는 간단한 문제이다.

 

[Python] collections 모듈의 Counter 사용법

 

먼저 입력된 파일들을 '.'를 기준으로 분리(split)하여 확장자 목록을 저장하고, Python에서 제공하는 라이브러리인 Counter를 사용하여 정답을 도출했다.

 

lambda 함수를 활용한 정렬, Counter의 사용법을 복습해볼 수 있었던 문제였다 :)