[프로그래머스 level2 파일명 정렬] lambda를 활용하여 쉽게 정렬하기

오늘도 삼성 기출문제를 도전하였지만,,너무 안풀렸다.

신기하게도 지금 작성하는 문제는 level2이지만 10분이 소요됐다..

곧 3단계로 가는데 3단계는 어마무시할 것 같아서 무섭다 ㅎㅎ

(벌써 2단계는 한 문제밖에 남지않았다..종강하고 꾸준히 푼 결과, 벌써 100문제 넘게 풀었다!)

 

해결해야하는 문제는 다음과 같았다.

programmers.co.kr/learn/courses/30/lessons/17686

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

 

 

문제설명↓

파일명 정렬

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

저장소 서버에는 프로그램의 과거 버전을 모두 담고 있어, 이름 순으로 정렬된 파일 목록은 보기가 불편했다. 파일을 이름 순으로 정렬하면 나중에 만들어진 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]

 

 

 

 

 

head를 문자순으로 정렬하고, number를 숫자순으로 정렬하는 파일 정리 알고리즘을 구현해야 했다.

우선 files로 들어온 것들을 head, number, tail로 구분하여 temp 배열에 저장하였다.

이는, 

현재 문자가 숫자인 경우, number에 추가하고

현재 문자 앞에 숫자가 없었고, 현재 문자가 숫자가 아니라면 head에 추가하고

숫자가 앞에 나왔었지만, 현재 문자가 숫자가 아닌 경우 뒷부분 전체를 tail에 저장하고 for문을 break 하도록 하였다.

if i[j].isdigit() :
	number.append(i[j])
	flag = 1
elif flag == 0 and not i[j].isdigit() :
	head.append(i[j])
elif flag == 1 and not i[j].isdigit() :
	tail = i[j:]
	break

 

 

 

temp에 저장할 때는 ''.join()을 활용하여 head, number, tail을 문자열 형태로 저장하였다.

temp.append([''.join(head),''.join(number),''.join(tail)])

 

 

 

여기서 sorted 함수를 통해 배열을 정렬하였는데, 함수 자리에 lambda를 활용하였다.

우선, head의 대소문자를 검사하지 않고 사전순으로 정렬하기 위해서 head에 해당하는 x[0]을 소문자로 하여 정렬하도록 하였다.

head에 해당하는 글자가 같은 경우, Number에 해당하는 x[1]을 int형으로 변환하여 숫자 순으로 정렬하도록 하였다.

temp = sorted(temp,key = lambda x : (x[0].lower(), int(x[1])))

 

 

정렬된 것들은 head,number,tail 배열을 하나의 문자열로 answer에 저장하도록 하였다.

이런 아이디어를 바탕으로 작성한 코드는 다음과 같다.

def solution(files):
    temp = []
    answer = []
    for i in files :
        head = []
        number = []
        tail = []
        flag = 0
        for j in range(len(i)):
            if i[j].isdigit() :
                number.append(i[j])
                flag = 1
            elif flag == 0 and not i[j].isdigit() :
                head.append(i[j])
            elif flag == 1 and not i[j].isdigit() :
                tail = i[j:]
                break

        temp.append([''.join(head),''.join(number),''.join(tail)])

    temp = sorted(temp,key = lambda x : (x[0].lower(), int(x[1])))

    for i in temp :
        answer.append(''.join(i))

    return answer

 

 

 

 

 

코드를 작성하면서 정규식을 활용하면 좋을 것 같다고 생각하였는데 역시나, 다른 사람의 코드에서 정규식을 활용한 코드가 있어 가져와보았다.

import re

def solution(files):
    a = sorted(files, key=lambda file : int(re.findall('\d+', file)[0]))
    b = sorted(a, key=lambda file : re.split('\d+', file.lower())[0])
    return b

이는 정규식을 활용하기 위해 import re를 하였고, '\d+'를 통해 숫자 여부를 검사하여 files에서 처음 등장하는 숫자(이 문제에서의 number)를 int형으로 하여, 순서대로 정렬하여 a에 담았다. b는 숫자를 기준으로 문자열을 나누었을 때 앞부분에 해당하는 문자(head에 해당)를 소문자로 하여 정렬해주었다. 이는 대소문자 구분 없이 정렬하기 위해 사용한 방법으로 나와 아이디어가 같았다.

 

나는 head로 정렬하고, number로 정렬하였지만, 이 풀이와 같이 한다면 숫자를 먼저 정렬하고 head로 정렬하였다는 것을 알 수 있다.

즉, 다양한 풀이가 존재할 수 있다는 것을 알게 해준다.

 

앞으로는 생각에 따라 자유자재로 코딩을 할 수 있었으면 좋겠다⭐️