[알고리즘 문제풀이] 프로그래머스 파일명정렬

오늘 푼 문제는 프로그래머스 - 파일명정렬 이다.
백준보다 프로그래머스가 개인적으로 더 편해서 다시 프로그래머스로 돌아왔다.
레벨2-4 / Java / 안 푼 문제로 필터링해서 아무 문제나 푸는 방법으로..

문제 설명

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

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

풀이 방법

문제 조건이 아주 명확해서 간단한 문제였다. 근데 문제가 진짜 실용적인 문제 해결을 다룬 문제라고 생각했다. 파일 정렬 진짜 좀 불편하지 않나요.. 저렇게 되면 좋겠는데 진짜..

어쨌든 TAIL은 전혀 정렬에 영향을 주지 않으므로, 첫 번째로 등장하는 숫자를 찾고 그것을 NUMBER, 그 이전까지의 문자열을 HEAD로 지정하고 정렬 우선순위를 HEAD > NUMBER로만 지정해주면 된다.

여기서 첫 번째로 등장하는 숫자를 찾는 것은 정규식을 이용했다. 그냥 String.matches는 존재 여부만 알 수 있는데, 나는 위치를 알아야했기 때문에 java.util.regex를 사용했다. 이건 처음 써봤는데 편리했다. group 속성으로 따로 substring을 해주지 않아도 됐다.

또한 정렬 우선 순위를 지정하기 위해서 나는 처음에는 File이라는 클래스를 만들고 Comparator를 확장하는 방식을 취하려고 했다. 하지만 그렇다면 입력으로 받은 문자열을 다 객체로 만들어서 배열에 넣어줘야하고, 정렬 후에도 다시 스트링으로 반환해야 하기 때문에 두 번의 Converting이 필요하게 된다. 그래서 나는 그냥 입력 문자열 배열 자체를 정렬하고, Comparator를 재정의하는 방식을 택했다.

사실 처음 생각했던 것처럼 파일 클래스를 만들어두고 각 속성과 정렬 기준을 안에 정의해두는 것이 더 직관적이고 가독성도 있는 좋은 코드라고 생각하기는 한다 ! 또한 나중에 정렬 기준이 바뀌거나 TAIL도 추가된다거나 파일 이름 형식이 바뀐다거나 등등 변경 사항이 있을 때 확장성을 생각해도 명확히 책임을 분리해두는 것이 좋다고 생각한다 ! 이건 오직 당장의 문제 풀이만을 생각한 풀이이다.

이렇게 되면 matcher와 pattern이 메모리를 많이 잡아먹지 않도록 잘 사용해야겠다는 생각이 들었다. 우선 pattern은 변하지 않으므로 static으로 선언해두고 사용하였다. 또한 matcher와 head/number를 저장할 변수도 두 객체를 비교하는 데 사용하기 위해서 각 두 개를 static으로 선언해두긴 했는데 사실 이건 메모리적으로 효율적인가 잘 모르겠고, compare override 함수에서 편하게 사용하기 위해서 이렇게 선언해두었다..^^..

어쨌든 이렇게 문제를 해결하였고, 코드는 간단하다 !

코드

import java.util.Arrays;
import java.util.Comparator;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

class Solution {
    static Pattern pattern = Pattern.compile("[0-9]+");
    static Matcher matcher1, matcher2;
    static String head1, head2;
    static int number1, number2;
    public String[] solution(String[] files) {
        Arrays.sort(files, new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                matcher1 = pattern.matcher(o1);
                matcher2 = pattern.matcher(o2);
                matcher1.find();
                matcher2.find();
                head1 = o1.substring(0, matcher1.start()).toLowerCase();
                head2 = o2.substring(0, matcher2.start()).toLowerCase();
                number1 = Integer.parseInt(matcher1.group());
                number2 = Integer.parseInt(matcher2.group());
                if(head1.equals(head2)){
                    return number1-number2;
                }
                return head1.compareTo(head2);
            }
        });
        return files;
    }
}