파일명 정렬 [Python]

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

문제 설명

• 세 차례의 코딩 테스트와 두 차례의 면접이라는 기나긴 블라인드 공채를 무사히 통과해 카카오에 입사한 무지는 파일 저장소 서버 관리를 맡게 되었다.
• 저장소 서버에는 프로그램의 과거 버전을 모두 담고 있어, 이름 순으로 정렬된 파일 목록은 보기가 불편했다. 파일을 이름 순으로 정렬하면 나중에 만들어진 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 부분을 기준으로 사전 순으로 정렬한다. 이때, 문자열 비교 시 대소문자 구분을 하지 않는다. MUZI와 muzi, MuZi는 정렬 시에 같은 순서로 취급된다.
• 파일명의 HEAD 부분이 대소문자 차이 외에는 같을 경우, NUMBER의 숫자 순으로 정렬한다. 9 < 10 < 0011 < 012 < 13 < 014 순으로 정렬된다. 숫자 앞의 0은 무시되며, 012와 12는 정렬 시에 같은 같은 값으로 처리된다.
• 두 파일의 HEAD 부분과, NUMBER의 숫자도 같을 경우, 원래 입력에 주어진 순서를 유지한다. MUZI01.zip과 muzi1.png가 입력으로 들어오면, 정렬 후에도 입력 시 주어진 두 파일의 순서가 바뀌어서는 안 된다.
• 무지를 도와 파일명 정렬 프로그램을 구현하라.

문제 풀이

• digit_index : 주어진 file에서 숫자로 시작하는 맨 처음의 index를 반환하는 함수
• number_index : 주어진 file에서 숫자가 아닌것으로 시작하는 맨 처음의 index를 반환하는 함수, 만일 모두 숫자라면 0을 반환함
• digit_index를 사용하여 head부분을 찾는다. 이후는 number + tail이므로, number_index 함수를 사용해서 number를 부분을 찾는다.
• 만일 tail이 없는 것도 가정하여 number_index 함수는 deafult로 0으로 둔다.
• head, number, tail을 new_files라는 리스트에 append 한다.
• 이후 sorted 함수를 사용하여 1) head, 2) number의 순으로 내림차순 해준뒤, head, number, tail을 다시 join 해주면끝, 여기서 head는 대소문자를 가리지 않는다고했으니 lower 함수로 일단 소문자로 일괄 변경한다.
• python의 sorted는 정렬 조건이 같다면 자동으로 index순으로 정렬을 하게 되므로, 굳이 index를 따로 저장해둘 필요는 없다.
• 나중에 알게된 사실이지만 head, number, tail을 나누는것을 그냥 정규표현식을 썻으면 더 간편하게 되었을것 같다.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

def digit_index(file):
    for s in file:
        if s.isdigit():
            idx = file.index(s)
            return idx
        
def number_index(file):
    idx = 0
    for s in file:
        if s.isdigit() == False:
            idx = file.index(s)
            break
    return idx
        
def solution(files):
    new_files = []
    for file in files:
        idx = digit_index(file)
        head = file[:idx]

        temp = file[idx:]
        number_idx = number_index(temp)

        if number_idx != 0:
            number = temp[:number_idx]
            tail = temp[number_idx:]
        else:
            number = temp
            tail = ''

        new_files.append([head, number, tail])

    sorts = sorted(new_files, key = lambda x : (x[0].lower(), int(x[1])))
    k = [''.join(sort) for sort in sorts]
    return k

1
2

files = ["F-5 Freedom Fighter", "B-50 Superfortress", "A-10 Thunderbolt II", "F-14 Tomcat"]
solution(files)

1
2
3
4

['A-10 Thunderbolt II',
 'B-50 Superfortress',
 'F-5 Freedom Fighter',
 'F-14 Tomcat']

1
2

files = ["img12.png", "img10.png", "img02.png", "img1.png", "IMG01.GIF", "img2.JPG"]
solution(files)

1

['img1.png', 'IMG01.GIF', 'img02.png', 'img2.JPG', 'img10.png', 'img12.png']

1
2

files = ["img12", "img10.png", "img02.png", "img1.png", "IMG01.GIF", "img2.JPG"]
solution(files)

1

['img1.png', 'IMG01.GIF', 'img02.png', 'img2.JPG', 'img10.png', 'img12']