[프로그래머스] [3차] 파일명 정렬

https://programmers.co.kr/learn/courses/30/lessons/17686

코딩테스트 연습 - [3차] 파일명 정렬

 

정규식을 이용하여 file 의 숫자를 기준으로 split 한다.

(\d+), 소괄호를 통해 숫자 형태의 문자열을 배열에 담아 [파일명(영문자), 파일명(숫자), .확장자] 로 전체 file 을 받아 file[0].lower() 과 int(file[1]) 로 오름차순으로 정렬한다.

 

import re

def solution(files):
    answer = []
    array = [re.split(r'(\d+)', file) for file in files]
    array.sort(key = lambda x : (x[0].lower(), int(x[1])))
    
    for i in array:
        answer.append(''.join(i)) 
    
    return answer

print(solution(['img12.png', 'img10.png', 'img02.png', 'img1.png', 'IMG01.GIF', 'img2.JPG'])) # ['img1.png', 'IMG01.GIF', 'img02.png', 'img2.JPG', 'img10.png', 'img12.png']