[python]프로그래머스 LEVEL2> 2018년 카카오 코딩테스트 - 파일명 정렬

안녕하세용! 오늘은 ,, 2018년 카카오코딩테스트문제를 가져왔습니다. 이 문제 테케를 다 통과하는데 10시간이상 걸렸는데.. 그 이유는 밑에 문제풀이에서 설명드리겠습니다. 하나를 고치니 테케 10개가 성공했다는,,!

그치만 다른 풀이 3-4줄 되는것을 보고 또!!!!!!!!!!  눈물을 흘리며 새로운 걸 배웠슴당,, 사실 아는 함수였는데 사용을 잘 안했더니 한번에 생각나질 않았네요!

역시 알고리즘은 충분한 연습과 습득이 되어야 제 시간안에 빠르게 풀 수 있는 것 같습니다.

여러분도 웬만하면 다른사람의 풀이는 보지않고 내가 최대한으로 해보고 그 이후에 코드를 비교해보며 공부를 하는 것을 정말정말 추천드립니다.!이렇게해야 문제갯수로 봤을 땐 적게 풀었을지 몰라도 실력향상은 빠르기에!!!

풀이를 3개를 가져왔는데 1개는 re라이브러리를 전혀쓰지 않은 풀이 다른 하나는 re라이브러리를 사용한 간단한 풀이입니다.  

오늘도 봐주셔서 감사합니당 :-)

 

 

문제 설명

세 차례의 코딩 테스트와 두 차례의 면접이라는 기나긴 블라인드 공채를 무사히 통과해 카카오에 입사한 무지는 파일 저장소 서버 관리를 맡게 되었다.

저장소 서버에는 프로그램의 과거 버전을 모두 담고 있어, 이름 순으로 정렬된 파일 목록은 보기가 불편했다. 파일을 이름 순으로 정렬하면 나중에 만들어진 ver-10.zip이 ver-9.zip보다 먼저 표시되기 때문이다.

버전 번호 외에도 숫자가 포함된 파일 목록은 여러 면에서 관리하기 불편했다. 예컨대 파일 목록이 ["img12.png", "img10.png", "img2.png", "img1.png"]일 경우, 일반적인 정렬은 ["img1.png", "img10.png", "img12.png", "img2.png"] 순이 되지만, 숫자 순으로 정렬된 ["img1.png", "img2.png", "img10.png", img12.png"] 순이 훨씬 자연스럽다.

무지는 단순한 문자 코드 순이 아닌, 파일명에 포함된 숫자를 반영한 정렬 기능을 저장소 관리 프로그램에 구현하기로 했다.

소스 파일 저장소에 저장된 파일명은 100 글자 이내로, 영문 대소문자, 숫자, 공백(" "), 마침표("."), 빼기 부호("-")만으로 이루어져 있다. 파일명은 영문자로 시작하며, 숫자를 하나 이상 포함하고 있다.

파일명은 크게 HEAD, NUMBER, TAIL의 세 부분으로 구성된다.

• HEAD는 숫자가 아닌 문자로 이루어져 있으며, 최소한 한 글자 이상이다.
• NUMBER는 한 글자에서 최대 다섯 글자 사이의 연속된 숫자로 이루어져 있으며, 앞쪽에 0이 올 수 있다. 0부터 99999 사이의 숫자로, 00000이나 0101 등도 가능하다.
• TAIL은 그 나머지 부분으로, 여기에는 숫자가 다시 나타날 수도 있으며, 아무 글자도 없을 수 있다.

 

파일명	HEAD	NUMBER	TAIL
foo9.txt	foo	9	.txt
foo010bar020.zip	foo	010	bar020.zip
F-15	F-	15	(빈 문자열)

 

파일명을 세 부분으로 나눈 후, 다음 기준에 따라 파일명을 정렬한다.

• 파일명은 우선 HEAD 부분을 기준으로 사전 순으로 정렬한다. 이때, 문자열 비교 시 대소문자 구분을 하지 않는다. MUZI와 muzi, MuZi는 정렬 시에 같은 순서로 취급된다.
• 파일명의 HEAD 부분이 대소문자 차이 외에는 같을 경우, NUMBER의 숫자 순으로 정렬한다. 9 < 10 < 0011 < 012 < 13 < 014 순으로 정렬된다. 숫자 앞의 0은 무시되며, 012와 12는 정렬 시에 같은 같은 값으로 처리된다.
• 두 파일의 HEAD 부분과, NUMBER의 숫자도 같을 경우, 원래 입력에 주어진 순서를 유지한다. MUZI01.zip과 muzi1.png가 입력으로 들어오면, 정렬 후에도 입력 시 주어진 두 파일의 순서가 바뀌어서는 안 된다.

무지를 도와 파일명 정렬 프로그램을 구현하라.

입력 형식

입력으로 배열 files가 주어진다.

• files는 1000 개 이하의 파일명을 포함하는 문자열 배열이다.
• 각 파일명은 100 글자 이하 길이로, 영문 대소문자, 숫자, 공백(" "), 마침표("."), 빼기 부호("-")만으로 이루어져 있다. 파일명은 영문자로 시작하며, 숫자를 하나 이상 포함하고 있다.
• 중복된 파일명은 없으나, 대소문자나 숫자 앞부분의 0 차이가 있는 경우는 함께 주어질 수 있다. (muzi1.txt, MUZI1.txt, muzi001.txt, muzi1.TXT는 함께 입력으로 주어질 수 있다.)

출력 형식

위 기준에 따라 정렬된 배열을 출력한다.

입출력 예제

입력: ["img12.png", "img10.png", "img02.png", "img1.png", "IMG01.GIF", "img2.JPG"]
출력: ["img1.png", "IMG01.GIF", "img02.png", "img2.JPG", "img10.png", "img12.png"]

입력: ["F-5 Freedom Fighter", "B-50 Superfortress", "A-10 Thunderbolt II", "F-14 Tomcat"]
출력: ["A-10 Thunderbolt II", "B-50 Superfortress", "F-5 Freedom Fighter", "F-14 Tomcat"]

 

문제풀이 1

 

def solution(files):
    s=[]
    answer=[]
    figure='0123456789'
    files_element=[i for i in enumerate(files)]
    for index,i in files_element:
        for a in range(len(i)):
            if i[a] in figure:
                s.append([index,i[:a].lower(),i[a:]])
                break            #[index값, head값,number+tail값]
    for index in range(0,len(s)):
        if len(s[index][-1])<6:
            s[index][-1]=s[index][-1].lstrip('0')
            for z in range(len(s[index][-1])):
                if s[index][-1][z] not in figure :
                    s[index][-1]=s[index][-1][0:z]
                    break
        else:
            s[index][-1]=s[index][-1][0:5].lstrip('0')
            for t in range(len(s[index][-1])):
                if s[index][-1][t] not in figure:
                    s[index][-1]=s[index][-1][0:t]
                    break
        if len(s[index][-1])==0:
            s[index][-1]='0'

        s[index][-1]=int(s[index][-1]) #전체 s값들의 number문자열을 정수로 바꿈        



    s.sort(key=lambda x:(x[1],x[2],x[0]))      # 정렬순서 head - number - 입력순 
    for i in range(len(s)):
        answer.append(files[s[i][0]])
    return answer

약간의 복잡함을 갖고있는 첫 번째 풀이입니다.

설명을 드리자면 

첫 번째 for문 →전체 s 리스트의 문자열에 대해서  [index, head, number+tail]로 나누는 코드

두 번째 for문 → number+tail 값의 길이에 대해서 6이상인 길이와 5이하인 길이를 if-else절로 나눠 number부분을 나오게한 코드 

★중요한 부분★

'0'도  number로 취급하기 때문에 길이 5개로 자르고 -> lstrip을 적용해야한다.

마지막 if절을 첨에 안넣어서 테케에서 많은 런타임에러가 났습니다.

그 이유는 위에서 lstrip('0')을 적용해줬기 때문에 만약 number가 0으로만 이루어져 있다면 number가 '' 즉 공백이 되어버리기 때문에 오류가 생깁니다. 

그래서 if절을 적용해주어야 합니다.

 

 

문제 풀이 2

 

import re
def solution(files):

    r = re.compile('([a-zA-Z- .]+)([0-9]+)')

    return sorted(files,key=lambda x: (r.findall(x)[0][0].lower(),int(r.findall(x)[0][1]),files.index(x)))

 

바로 이 두 번째 풀이가 정말 파이썬 다운풀이이다. 이 두줄 코드로 파이썬의 장점을 표현하는듯하다 ㅎㅎ

요기서 !! 중요하게 봐야할점은 findall의 기능을 아주유심히 잘봐야한다! 

' 아 findall 을쓰면 이렇게 필요한 부분을 걸러내주는구나' 을 넘어 어떤식으로 입력 값을 넣었을때 도출이 다른지를 확인해보는 것을 추천드립니다.

예를들어, compile값에서 ''안에 들어있는 ()값에 들어있는 값에 맞는 문자열이 튜플 값으로 나온다는 점!

그래서 return 값에서 findall에서앞에 [0]을 해준다는 것을 알아두시길 바랍니다!

헷갈리시는 분들을 위해 예제 추가합니다.

 so='img12301.jpg123'
>>> r=re.compile('([a-z- .]+)([0-9]+)')
>>> r.findall(so)
[('img', '12301'), ('.jpg', '123')]

이렇게 ' ' 쉼표안에있는 값들의 단위로 끊어줍니다.

>>> r=re.compile('([a-z- .]+)([0-9]+)([a-z- .]+)')
>>> r.findall(so)
[('img', '12301', '.jpg')]

>>> r=re.compile('([a-z-0-9 .]+)')
>>> r.findall(so)
['img12301.jpg123']

요 위 예제를 보셨다면 compile과 findall을 확실히 어떻게 쓰면 될지 바로 아실겁니다.

 

 

포스팅은 여기까지입니다

 

감사합니다 ^-^ ♡