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

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

프로그래머스

 

* 내 풀이 (시간 초과):

def solution(files):
    temp = []
    for file in files:
        i, size = 1, 0
        while True:
            if file[i].isdigit():
                while file[i:i + size + 1].isdigit():
                    size += 1
                break
            i += 1
        temp.append((file[:i], file[i:i + size], file[i + size:]))

    return list(map(''.join, sorted(temp, key = lambda x:(x[0].upper(), int(x[1])))))

• 아이디어: 각 파일별로 최초로 나오는 숫자부터 연속되는 숫자들까지의 인덱스를 활용하여 구분하고 문제의 조건에 맞게 파일 정렬
• 파일 갯수와 각 파일의 문자열 길이가 크지 않고 연속 숫자 갯수도 최대 5 밖에 되지 않는데 왜 초과되는지 모르겠음

* 개선한 풀이:

def solution(files):
    ans = []
    for file in files:
        temp, check_num = [''] * 3, False
        for i, char in enumerate(file):
            if char.isdigit():
                temp[1] += char
                check_num = True
            elif check_num == False:
                temp[0] += char
            else:
                temp[2] = file[i:]
                break

        ans.append(temp)
        
    return list(map(''.join, sorted(ans, key = lambda x:(x[0].upper(), int(x[1])))))

• for 문 돌면서 하나하나 ans에 입력하는 방식