[Python] 프로그래머스 코딩테스트 연습 level 2 (파일명 정렬)

1. 문제 설명

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

저장소 서버에는 프로그램의 과거 버전을 모두 담고 있어, 이름 순으로 정렬된 파일 목록은 보기가 불편했다. 파일을 이름 순으로 정렬하면 나중에 만들어진 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은 그 나머지 부분으로, 여기에는 숫자가 다시 나타날 수도 있으며, 아무 글자도 없을 수 있다.

 

 

 

 

입력 형식

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

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

출력 형식

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

 

 

2. 내 풀이

- 정확도 : 55%

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

 

tail에 숫자가 존재할수 있음

 

import re
def solution(files):
    answer = []
    
    for i in files:
        number = re.findall('\d+', i)
        word = re.split('\d+', i)
        
        if len(word) <=1:
            answer.append([word[0], number[0]])
        elif len(number)>=2:
            answer.append([word[0], number[0],i.replace(word[0],'',1).replace(number[0],'',1)])
        else:
            answer.append([word[0], number[0], word[1]])
        
    answer.sort(key=lambda x:(x[0].lower(),int(x[1])))
    answer = [''.join(x) for x in answer]
    return answer

 

1.  re.findall()함수를 사용해 모든 숫자 추출

2. 숫자를 기준으로 문자열을 split() 해서 리스트로 반환

3. tail이 없는 경우 (len(word)<=1) : word[0]와 number[0] answer에 추가

tail에 숫자가 존재하는 경우 (len(number)>=2): 문자열에서 word[0]와 number[0]를 공백으로 변환 후 추가

4. 대, 소문자 구분하지 않으므로 split된 문자열 리스트를 기준으로 head부분은 모두 소문자로 변환, number부분은 정수 변환 후 정렬

5. 모든 리스트를 join()함수를 사용하여 합친 후 반환

 

 

3. 다른 사람 풀이

import re

def solution(files):
    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

1. 문자열에서 숫자 추출 후 정렬

2. 숫자를 기준으로 문자열 분할 후 리스트의 첫번째 요소를 소문자로 변환하여 정렬