[알고리즘] 백준 20291 파이썬 - 파일 정리

 

20291번: 파일 정리

문제

친구로부터 노트북을 중고로 산 스브러스는 노트북을 켜자마자 경악할 수밖에 없었다. 바탕화면에 온갖 파일들이 정리도 안 된 채 가득했기 때문이다. 그리고 화면의 구석에서 친구의 메시지를 확인할 수 있었다.

바탕화면의 파일들에는 값진 보물에 대한 정보가 들어 있어. 하나라도 지우게 된다면 보물은 물론이고 다시는 노트북을 쓸 수 없게 될 거야. 파일들을 잘 분석해서 보물의 주인공이 될 수 있길 바랄게. 힌트는 “확장자”야.

화가 났던 스브러스는 보물 이야기에 금세 화가 풀렸고 보물의 정보를 알아내려고 애썼다. 하지만 파일이 너무 많은 탓에 이내 포기했고 보물의 절반을 보상으로 파일의 정리를 요청해왔다. 스브러스의 요청은 다음과 같다.

• 파일을 확장자 별로 정리해서 몇 개씩 있는지 알려줘
• 보기 편하게 확장자들을 사전 순으로 정렬해 줘

그럼 보물의 절반을 얻어내기 위해 얼른 스브러스의 노트북 파일 정리를 해줄 프로그램을 만들자!

 

 

문제풀이

import sys
input = sys.stdin.readline
n=int(input())
ans={}
#01
for i in range(n):
    a,b = map(str,input().rstrip().split('.'))
    if b in ans:
        ans[b]+=1
    else:
        ans[b]=1
#02
ans=sorted(ans.items(),key=lambda x:x[0])
#03
for i in ans:
    print(i[0] ,i[1])

- #01 : "."을 기준으로 왼쪽은 파일이름, 오른쪽은 확장자이다. 그렇기에 "."을 기준으로 나눠준다.

           딕셔너리 ans에 b라는 확장자가 있다면 1를 더해주고, 없다면 새로 만들어준다.

- #02 : ans.items()의 첫 번째 값을 기준으로 오름차순 정렬해준다.

- #03 : 정렬된 리스트 ans를 for문을 통해 문제 형식에 맞게 출력해준다.