[Programmers] 파일명 정렬

 

 

 

 

해설 코드 

 

def solution(files):
    ans = []
    head, number, tail = '','',''
    
    for file in files:
        for i in range(len(file)):
            # 우선 파일을 head와 number로만 나눈다. 
            if file[i].isdigit():
                head = file[:i]
                number = file[i:]
                
                # number을 다시 number와 tail로 나눈다. 
                for k in range(len(number)):
                    if not number[k].isdigit():
                        tail = number[k:]
                        number = number[:k]
                        break 
                
                # 배열에 각각의 값을 추가한 후 head,number,tail값을 reset한다. 
                ans.append([head,number,tail])
                head,number,tail = '', '', ''
                break
    
    # 대소문자 구분을 하지 않기 때문에 소문자로 바꾼 값 기준으로 정렬 후 숫자의 값을 비교한다. 
    ans.sort(key = lambda x: (x[0].lower(),int(x[1])))
    return [''.join(x) for x in ans]

분석 > 풀긴 풀었으나 시간초과가 난 문제. 해설 코드에서 얻은 좋은 아이디어는 애초에 head와 number 부분을 file내 숫자를 기준으로 분리하고 number을 다시 진짜 number와 tail로 나누어 값을 얻은 것이다. 나의 경우 head와 number을 구한 후 원래 file 값에서 head와 number값을 공백으로 replace하는 방법을 선택했었는데, 이 코드를 보니 그냥 처음부터 두 개로 분리 후 다시 분리하는 방법을 택하면 훨씬 효율적으로 문제를 해결할 수 있다는 것을 알게 되었다. 또 대소문자를 구분하지 않고 정렬한 후 숫자를 기준으로 정렬하는 방법을 알게 되었는데, 대소문자를 구분하지 않는다는 것은 문자를 그냥 소문자로 바꾼 값을 기준으로 정렬한다는 뜻이기 때문에 .lower을 이용해 정렬해주고 현재 각 값은 문자열이기 때문에 숫자 비교를 위해 정수형으로 바꿔준 값을 기준으로 정렬하면 내가 원하는 값을 얻을 수 있었다. 

 

 

 

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