링크

문제 링크

문제 설명

파일명 정렬

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

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

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

1. 파일명은 우선 HEAD 부분을 기준으로 사전 순으로 정렬한다. 이때, 문자열 비교 시 대소문자 구분을 하지 않는다. MUZI와 muzi, MuZi는 정렬 시에 같은 순서로 취급된다.

2. 파일명의 HEAD 부분이 대소문자 차이 외에는 같을 경우, NUMBER의 숫자 순으로 정렬한다. 9 < 10 < 0011 < 012 < 13 < 014 순으로 정렬된다. 숫자 앞의 0은 무시되며, 012와 12는 정렬 시에 같은 같은 값으로 처리된다.

3. 두 파일의 HEAD 부분과, NUMBER의 숫자도 같을 경우, 원래 입력에 주어진 순서를 유지한다. MUZI01.zip과 muzi1.png가 입력으로 들어오면, 정렬 후에도 입력 시 주어진 두 파일의 순서가 바뀌어서는 안 된다.

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

제한 사항

HEAD는 숫자가 아닌 문자로 이루어져 있으며, 최소한 한 글자 이상이다.

NUMBER는 한 글자에서 최대 다섯 글자 사이의 연속된 숫자로 이루어져 있으며, 앞쪽에 0이 올 수 있다. 0부터 99999 사이의 숫자로, 00000이나 0101 등도 가능하다.

TAIL은 그 나머지 부분으로, 여기에는 숫자가 다시 나타날 수도 있으며, 아무 글자도 없을 수 있다.

입력 형식

입력으로 배열 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. 문제에서 친히 제시해준 1번, 2번, 3번 정렬을 따라가자
2. get_index1 메소드를 통해 각 문자열 HEAD가 끝나는 지점의 index를 얻는다
3. 얻은 index로 substring을 구하고 소문자로 바꾼다.
4. 둘의 사전순 정렬이 똑같다면 get_index2로 NUMBER가 끝나는 지점의 index를 얻는다.
5. 얻은 index로 substring을 Integer로 형변환한 다음 값을 비교한다.

문제를 보자마자 Arrays.sort()에서 오버라이드하여 코드가 좀 복잡해지더라도 한번에 풀고싶었다.

그러나 문제에서 Head, Number, Tail 세 부분으로 친히 나누는 것을 보고 자꾸 저 과정을 거쳐서 해야하나..? 라는 고민에 빠졌었다.

그러나 아무리 생각해도 저 풀이는 불필요하게 시간, 공간만 더 잡아먹는다는 판단을 하였다.

저 방식대로 Head를 기준으로 정렬을 했다치자, 그 상태를 어디다 담은 채로 Number를 기준으로 두 번째 정렬을 해야하고 그것이 결국 stable 해야하니 어질어질했다.

그래서 정렬 메소드를 오버라이드하기로 하였다. 그 안에서 코드를 좀 깔끔하게 가독성있게 하고싶어서 get_index1과 get_index2를 선언하였다.

런타임에러를 겪긴했지만 얼마 지나지않아 문제는 풀렸다!

다른 사람들의 풀이에 비해 코드가 짧은 것을 보니 내심 만족스러웠다.

코드

import java.util.*;
class Solution {
    public int get_index1(String s){
        int i;
        for(i = 0; i < s.length() && (s.charAt(i) < '0' || s.charAt(i) > '9'); i++);
        return i;
    }
    public int get_index2(String s, int index){
        int i, len;
        for(i = index, len = 1; i < s.length() && s.charAt(i) >= '0' && s.charAt(i) <= '9'; i++, len++){
            if(len == 5)
                return i;
        }
        return i;
    }
    public String[] solution(String[] files) {
        String[] answer = {};
        Arrays.sort(files, new Comparator<String>(){
            @Override
            public int compare(String head1, String head2){
                int i, j;
                i = get_index1(head1);
                j = get_index1(head2);
                String s1 = head1.substring(0,i).toLowerCase();
                String s2 = head2.substring(0,j).toLowerCase();
                if(s1.toLowerCase().compareTo(s2.toLowerCase()) == 0){
                    String s3 = head1.substring(i,get_index2(head1,i+1));
                    String s4 = head2.substring(j,get_index2(head2,j+1));
                    if(Integer.parseInt(s3) > Integer.parseInt(s4))
                        return 1;
                    else if(Integer.parseInt(s3) == Integer.parseInt(s4))
                        return 0;
                    else 
                        return -1;
                }
                else 
                    return s1.compareTo(s2);
            }
        });
        answer = files;
        return answer;
    }
}