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

1. 문제 설명

문제 링크

• 입력 : 파일이름들
• 출력 : 문자부분, 숫자부분, 남은 부분으로 나누어서 정렬기준에 맞게 정렬된 파일명 출력

2. 코드

solution1.py

from functools import cmp_to_key

def compare(s1,s2):
    head1, number1, tail1 = s1[0], s1[1], s1[2]
    head2, number2, tail2 = s2[0], s2[1], s2[2]

    head1, head2 = head1.upper(), head2.upper() #알바벳 구분 없으니까
    number1, number2 = int(number1), int(number2) #비교를 위해 int화

    if head1 != head2:
        return -1 if head1 < head2 else 1 #만약 head1이 head2보다 작으면 교환(-1)
    else: #head1 == head2
        return number1 - number2 if number1 != number2 else 1 #숫자가 서로 같지 않을 때만 빼기(둘이 같을 때(둘을 서로 빼면 0이니까) 교환이 되어버림을 방지)

def split_fname(filename):
    HEAD = ""
    NUMBER = ""
    TAIL = ""
    i = 0
    while not filename[i].isnumeric():
        i+=1
    HEAD = filename[:i]

    j = i
    while filename[j].isnumeric():
        j+=1
        if j == len(filename): #TAIL이 없는 경우
            NUMBER = filename[i:j]
            return [HEAD, NUMBER, '']

    NUMBER = filename[i:j]
    TAIL = filename[j:]

    return [HEAD, NUMBER,TAIL]


def solution(files):
    answer = []
    filenames  = []
    for file in files: #파일명 하나씩 조회
        filenames.append(split_fname(file))

    filenames.sort(key=cmp_to_key(compare))

    for i, filename in enumerate(filenames):
        filenames[i] = ''.join(filename)
    return filenames

• from functools import cmp_to_key로 선언
• functools.cmp_to_key(func) 함수는 sorted와 같은 정렬 함수의 key 매개변수에 함수(func)를 전달할 때 사용하는 함수이다.
• 단, func 함수는 두 개의 인수를 받아들이고, 첫번째 인수를 기준으로 그들을 비교하여, 작으면 음수, 같으면 0, 크면 양수를 반환하는 비교 함수이어야 한다.
• 정렬을 할 때, cmp_to_key를 활용해야한다.
  • from functools import cmp_to_key로 선언
  • cmp_to_key안에는 compare 함수를 넣는다.
  • compare 함수는 비교해서 앞 원소가 더 작으면 -1을 반환해서 서로 교환(위치를 바꿈)하게하고,서로 같으면 0을 반환(교환), 앞 원소가 더 크면 1을 반환(교환하지않음)한다.
• 주의 1) tail이 없는 경우
  • tail이 없는 경우는 예외처리를 해주어야한다. 안하면 런타임 에러가 날 수 있음
• 주의 2) 파일명 원상복귀
  • ''.join(filename) 로 나누어놓은 파일명리스트를 string형태로 바꿔주자

solution2.py

def solution(files):
    tmp = []
    head, number, tail = '', '', ''

    for file in files:
        for i in range(len(file)):
            if file[i].isdigit():     # 숫자가 나오면 그 이전은 무조건 HEAD, 이후는 NUMBER, TAIL로 다시 구분
                head = file[:i]
                number = file[i:]

                for j in range(len(number)):    # NUMBER와 TAIL 구분 (숫자 안나오면 TAIL)
                    if not number[j].isdigit():
                        tail = number[j:]
                        number = number[:j]
                        break

                tmp.append([head, number, tail])
                head, number, tail = '', '', ''
                break

    tmp = sorted(tmp, key=lambda x:(x[0].lower(), int(x[1])))

    return [''.join(i) for i in tmp]

• for문과 break문 사용을 통해 쉽게 푼 풀이이다.
• 훨씬 간결하다.

solution3.py

import re
import re
def solution(files):
    temp = [re.split(r"([0-9]+)", file) for file in files]
    sort = sorted(temp, key = lambda x: (x[0].lower(), int(x[1])))

    return [''.join(s) for s in sort]

• 정규표현식을 사용하여 file의 숫자를 기준으로 split
• r'(\d+)' 도 사용가능

4. 참고

https://www.youtube.com/watch?v=IcNPpsG8i2s
https://wikidocs.net/109303
https://velog.io/@sem/%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%98%EB%A8%B8%EC%8A%A4-LEVEL2-%ED%8C%8C%EC%9D%BC%EB%AA%85-%EC%A0%95%EB%A0%AC-Python