세 차례의 코딩 테스트와 두 차례의 면접이라는 기나긴 블라인드 공채를 무사히 통과해 카카오에 입사한 무지는 파일 저장소 서버 관리를 맡게 되었다.
저장소 서버에는 프로그램의 과거 버전을 모두 담고 있어, 이름 순으로 정렬된 파일 목록은 보기가 불편했다. 파일을 이름 순으로 정렬하면 나중에 만들어진 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 |
---|---|---|
foo9.txt | foo 9 | .txt |
foo010bar020.zip | foo 010 | bar020.zip |
F-15 | F- 15 | (빈 문자열) |
파일명을 세 부분으로 나눈 후, 다음 기준에 따라 파일명을 정렬한다.
무지를 도와 파일명 정렬 프로그램을 구현하라.
입력으로 배열 files가 주어진다.
위 기준에 따라 정렬된 배열을 출력한다.
입출력 예제
입력: ["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"]
처음 숫자를 만났을 때와 숫자가 아닌 것을 만났을 때 적절한 코드를 작성한 후, 반복문을 종료시키는 break를 작성해야 하는 것에 주의한다.
정렬 시 우선순위는 첫번 째로 head 부분이고 그 다음에는 number인 것에 주의한다.
✍ 소스 코드
def solution(files):
answer = []
head, number, tail = '','',''
for file in files:
for i in range(len(file)):
if file[i].isdigit(): # 처음 숫자를 만나게 되면
head = file[:i] # 숫자를 만나기 이전까지 head
number = file[i:] # 숫자를 만난 것부터 number
for j in range(len(number)): # 처음 숫자를 만난 이후에
if not number[j].isdigit(): # 숫자가 아닌 것을 만나게 되면
tail = number[j:]
number = number[:j]
break # 숫자 아닌 것을 만나면 바로 if문 종료
answer.append([head, number, tail])
head, number, tail = '','',''
break
answer = sorted(answer, key=lambda x:(x[0].lower(), int(x[1])))
return [''.join(i) for i in answer]