[프로그래머스 Lv2,python]- 파일명

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

프로그래머스

def f(files): # head,number,tail 나누기
    head,number,tail=[],[],[]
    for name in files:
        idx,cnt=-1,0
        for i in range(len(name)): 
            if name[i].isdigit(): # 숫자이면
                if idx==-1: # 처음 나온 숫자일때
                    idx=i # 인덱스 저장해주기
                cnt+=1 # 숫자 몇자리인지
            if name[i]==".": # .뒤에도 숫자가 있을수있으므로 멈춰주기
                break
        head.append(name[:idx]) # 숫자 전까지
        number.append(name[idx:idx+cnt]) # 숫자
        tail.append(name[idx+cnt:]) # 숫자 이후
    return head,number,tail

def solution(files):
    le=len(files) 
    head,number,tail=f(files)
    
    for i in range(le-1):
        for j in range(le-1):
            if head[j].lower()>head[j+1].lower(): # 먼저 파일 이름과 비교하여 정렬
                head[j],head[j+1]=head[j+1],head[j]
                number[j],number[j+1]=number[j+1],number[j]
                tail[j],tail[j+1]=tail[j+1],tail[j]
            elif head[j].lower()==head[j+1].lower(): # 파일 이름이 같으면 숫자비교하기
                if int(number[j])> int(number[j+1]):
                    head[j],head[j+1]=head[j+1],head[j]
                    number[j],number[j+1]=number[j+1],number[j]
                    tail[j],tail[j+1]=tail[j+1],tail[j]
    for i in range(le):
        files[i]=head[i]+number[i]+tail[i] # 합쳐주기
    return files

위의 풀이는 버블정렬을 사용하여 푼 내 코드이고, 아래는 다른분이 푼 코드이다.

def head_compare(data):    
    route = 0
    answer = ['', '']
    for i, v in enumerate(data):
        if (route == 0): 
            if (v.isnumeric() == True): # 숫자일때
                answer[0] = data[:i].lower() # 이전까지는 파일 이름
                route = 1
                answer[1] += v
        elif (route == 1): # 숫자일때
            if (v.isnumeric() == True):
                answer[1] += v
            else: # 숫자가 아니면 끝내기
                break
    answer[1] = int(answer[1]) # int형으로 변환해주기
    return answer


def solution(files):
    answer = []

    files.sort(key = lambda x:head_compare(x)) # answer을 기준으로 정렬하기
    return files