[Programmers] 2018 Kakao > 파일명 정렬

문제. 파일명 정렬

---

세 차례의 코딩 테스트와 두 차례의 면접이라는 기나긴 블라인드 공채를 무사히 통과해 카카오에 입사한 무지는 파일 저장소 서버 관리를 맡게 되었다.

저장소 서버에는 프로그램의 과거 버전을 모두 담고 있어, 이름 순으로 정렬된 파일 목록은 보기가 불편했다. 파일을 이름 순으로 정렬하면 나중에 만들어진 ver-10.zip이 ver-9.zip보다 먼저 표시되기 때문이다.

버전 번호 외에도 숫자가 포함된 파일 목록은 여러 면에서 관리하기 불편했다.

예컨대 파일 목록이 [img12.png, img10.png, img2.png, img1.png]일 경우, 일반적인 정렬은 [img1.png, img10.png, img12.png, img2.png] 순이 되지만, 숫자 순으로 정렬된 [img1.png, img2.png, img10.png, img12.png"] 순이 훨씬 자연스럽다.

무지는 단순한 문자 코드 순이 아닌, 파일명에 포함된 숫자를 반영한 정렬 기능을 저장소 관리 프로그램에 구현하기로 했다.

소스 파일 저장소에 저장된 파일명은 100 글자 이내로, 영문 대소문자, 숫자, 공백(" ), 마침표(.), 빼기 부호(-")만으로 이루어져 있다. 파일명은 영문자로 시작하며, 숫자를 하나 이상 포함하고 있다.

 

파일명은 크게 HEAD, NUMBER, TAIL의 세 부분으로 구성된다.

• HEAD는 숫자가 아닌 문자로 이루어져 있으며, 최소한 한 글자 이상이다.
• NUMBER는 한 글자에서 최대 다섯 글자 사이의 연속된 숫자로 이루어져 있으며, 앞쪽에 0이 올 수 있다. 0부터 99999 사이의 숫자로, 00000이나 0101 등도 가능하다.
• TAIL은 그 나머지 부분으로, 여기에는 숫자가 다시 나타날 수도 있으며, 아무 글자도 없을 수 있다.

파일명	HEAD	NUMBER	TAIL
foo9.txt	foo	9	.txt
foo010bar020.zip	foo	010	bar020.zip
F-15	F-	15	(빈 문자열)

파일명을 세 부분으로 나눈 후, 다음 기준에 따라 파일명을 정렬한다.

• 파일명은 우선 HEAD 부분을 기준으로 사전 순으로 정렬한다. 이때, 문자열 비교 시 대소문자 구분을 하지 않는다. MUZI와 muzi, MuZi는 정렬 시에 같은 순서로 취급된다.
• 파일명의 HEAD 부분이 대소문자 차이 외에는 같을 경우, NUMBER의 숫자 순으로 정렬한다. 9 < 10 < 0011 < 012 < 13 < 014 순으로 정렬된다. 숫자 앞의 0은 무시되며, 012와 12는 정렬 시에 같은 같은 값으로 처리된다.
• 두 파일의 HEAD 부분과, NUMBER의 숫자도 같을 경우, 원래 입력에 주어진 순서를 유지한다. MUZI01.zip과 muzi1.png가 입력으로 들어오면, 정렬 후에도 입력 시 주어진 두 파일의 순서가 바뀌어서는 안 된다.

무지를 도와 파일명 정렬 프로그램을 구현하라.

입력 형식

입력으로 배열 files가 주어진다.

• files는 1000 개 이하의 파일명을 포함하는 문자열 배열이다.
• 각 파일명은 100 글자 이하 길이로, 영문 대소문자, 숫자, 공백(" ), 마침표(.), 빼기 부호(-")만으로 이루어져 있다. 파일명은 영문자로 시작하며, 숫자를 하나 이상 포함하고 있다.
• 중복된 파일명은 없으나, 대소문자나 숫자 앞부분의 0 차이가 있는 경우는 함께 주어질 수 있다. (muzi1.txt, MUZI1.txt, muzi001.txt, muzi1.TXT는 함께 입력으로 주어질 수 있다.)

출력 형식

위 기준에 따라 정렬된 배열을 출력한다.

입출력 예제

입력: [img12.png, img10.png, img02.png, img1.png, IMG01.GIF, img2.JPG]
출력: [img1.png, IMG01.GIF, img02.png, img2.JPG, img10.png, img12.png]

입력: [F-5 Freedom Fighter, B-50 Superfortress, A-10 Thunderbolt II, F-14 Tomcat]
출력: [A-10 Thunderbolt II, B-50 Superfortress, F-5 Freedom Fighter, F-14 Tomcat]

 

출처 : 프로그래머스 Level 2 파일명 정렬

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

---

나의 풀이

import re
def solution(files):
    answer = []
    # Head : 숫자가 아닌 문자 / 대소문자 구분 x / 문자열 오름차순
    # Number는 only 숫자로 이루어져 있음 / 숫자 오름차순 단, 09 < 10 <0011 앞의 0은 무시 따라서 int 처리 필요
    # 나머지 부분 / 원래 순서 유지
    split_fs = []
    for i in files:
        split_fs.append(re.split(r"([0-9]+)", i))
    
    # split_fs = [re.split(r"([0-9]+)", i) for i in files]
    
    #다중 조건으로 정렬하기
    sort_fs = sorted(split_fs, key = lambda x: (x[0].lower(),int(x[1])))
    
    for j in sort_fs:
        answer.append(''.join(j))
    return answer

파일명이 HEAD, NUMBER, TAIL로 구분이 되는데, 구분의 기준이 숫자이기에 정규표현식을 사용하기 위해서 re 모듈을 활용했습니다. 

먼저, 중간에 있고 오로지 숫자로 이루어져 있는 number를 기준으로 분리하면 [head / number / tail] 이런 식으로 나눌 수 있어 re.split을 활용했습니다. 프로그래밍을 한다면, 다양한 종류의 문자열 집합을 만나고 다뤄야 하기에 re 모듈은 반드시 알아둬야할 것 같습니다.

그 다음, sorted, key, lambda 구문을 활용하여 다중 조건의 정렬을 해결했습니다. 비교할 아이템이 여러 개일 경우, 해당 구문을 활용하면, 튜플로 결과값을 받을 수 있습니다. 여러 가지로 활용도가 높은 구문이기에 따로 메모해서 외웠습니다.

마지막으로, 튜플 상태의 sort_fs 값을 join을 활용하여 return하여 풀었습니다.

 

다른 사람의 풀이

import re

def solution(files):
	#\d는 숫자를 의미. [0-9]와 동일
    a = sorted(files, key=lambda file : int(re.findall('\d+', file)[0]))
    b = sorted(a, key=lambda file : re.split('\d+', file.lower())[0])
    return b

간결하게 파이썬을 잘 활용한 풀이라 가져왔습니다. 핵심은 먼저 number 부분을 정렬한 리스트를 head 부분으로 정렬한 부분입니다. key 부분에 re를 활용하였기에 따로 join할 필요 없이 return 가능하므로 파이썬을 아주 잘 활용한 풀이라고 생각했습니다.

 

참고할 부분

문제를 풀면서 간단하게 해결할 수 있는 모듈이나 패키지가 있을까 궁금해서 찾아봤습니다.

natsort 라이브러리는 텍스트로 된 숫자를 정렬하는 방법으로, natsorted()를 이용하면 아주 쉽게 텍스트 파일을 정렬할 수 있습니다.

import os
import natsort

a = ["img12.png", "img10.png", "img02.png", "img1.png", "IMG01.GIF", "img2.JPG"]
b = natsort.natsorted(a)
# b : ['IMG01.GIF', 'img1.png', 'img2.JPG', 'img02.png', 'img10.png', 'img12.png']

단점이 있다면, 01과 1, 02와 2를 같은 숫서로 보지 않는 점이 아쉽지만 re.split을 활용하면 해결할 수 있습니다. 많은 사람들이 쉽게 사용하는 라이브러리라 알아두면 좋을 것 같습니다.

 

출처 : https://natsort.readthedocs.io/en/master/api.html#standard-api