프로그래머스) 파일명 정렬

문제 : https://programmers.co.kr/learn/courses/30/lessons/17686?language=python3 

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

문제가 너무 길어 링크로..

 

 

header와 number부분을 확실하게 나누기만 하면 정렬을 해서 쉽게 풀 수 있는 문제다.

def solution(files):
    answer = []

    for i in range(len(files)):
        n_idx_s = -1
        n_idx_e = -1
        for j in range(len(files[i])):
            if files[i][j].isdigit():
                if n_idx_s < 0:
                    n_idx_s = j
            else:
                if n_idx_s >= 0 and n_idx_e < 0:
                    n_idx_e = j
                    break
        if n_idx_e < 0:
            n_idx_e = len(files[i])

        header = files[i][:n_idx_s]
        number = files[i][n_idx_s:n_idx_e]
        tail = files[i][n_idx_e:]

        answer.append([header,number,tail])

    answer.sort(key=lambda x:(x[0].lower(),int(x[1])))

    for i in range(len(answer)):
        answer[i] = "".join(answer[i])

    return answer

 

이 부분이 문제의 핵심이라고 생각한다. 

answer.sort(key=lambda x:(x[0].lower(),int(x[1])))

해당 링크의 내용을 응용했다.

https://ojy9612.tistory.com/33

python) 정렬에 lambda함수 활용하기