[3차] 파일명 정렬(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"]

---

✏️풀이

코드

import java.util.*;

class Solution {
	// 비교하기 편하도록 가공하는 메서드
    public String[] detach(String file) {
		// 헤더, 숫자, 외의 나머지
        String head = "", number = "", tail = "";
		// 인덱스
        int index = 0;
        
		// 파일의 길이만큼 반복
        for(; index < file.length(); index++) {
            char c = file.charAt(index);
			// 만약 숫자가 들어온다면 반복문 종료
            if(c >= '0' && c <= '9') {
                break;
            }
			// 아니라면 head에 저장
            head += file.charAt(index);
        }
        
		// 종료된 위치부터 파일의 길이만큼 반복
        for(; index < file.length(); index++) {
            char c = file.charAt(index);
			// 만약 숫자가 아닌 다른 것이 들어온다면 반복문 종료
            if(c < '0' || c > '9') {
                break;
            }
			// 숫자라면 number에 저장
            number += file.charAt(index);
        }
        
		// 외의 나머지는 tail에 저장
        tail = file.substring(index);
        
		// 나눠준 값을 반환
        return new String[] {head.toLowerCase(), number, tail};
    }

    public String[] solution(String[] files) {
		// 원하는대로 정렬하도록 수정
        Arrays.sort(files, new Comparator<String>() {
           @Override
            public int compare(String s1, String s2) {
                String[] file1 = detach(s1);
                String[] file2 = detach(s2);
                
				// head를 비교해서 사전 순대로 정렬
                int headValue = file1[0].compareTo(file2[0]);
                
				// head가 같다면 number를 비교해서 정렬
                if(headValue == 0) {
                    int number1 = Integer.parseInt(file1[1]);
                    int number2 = Integer.parseInt(file2[1]);
                    
                    return number1 - number2;
                } else {
                    return headValue;
                }
            }
        });
        
        return files;
    }
}

설명

데이터의 가공과 정렬을 통해 문제를 해결할 수 있었다. 이 가공과 정렬을 하는 과정이 쉽지 않았다..

가공은 detach 메서드를 만들어서 사용하였다. files 배열의 값 하나를 매개변수로 넘겨주면 head, number, tail로 나누어서 String 배열로 반환해주는 코드인데, for문에 초깃값을 설정해주는 것이 아니라 index 변수를 하나 선언해서 위의 변수만으로 위치를 옮겨가며 값을 나눠준다. head 뒤에 number 뒤에 tail이 나오기 때문에 굳이 처음부터 반복을 진행할 필요가 없기 때문이다.

solution 메서드에서 Arrays.sort를 사용하는데 이때 new Comparator을 같이 사용한다.
Comparator는 두 매개변수 객체를 비교할 때 사용한다. 이와 비슷하게 Comparable이 있다. 비슷하지만 차이점이 분명히 존재하는데 이 부분은 내가 헷갈릴 때 찾아보는 Comparable과 Comparator의 이해에서 자세한 설명을 읽어보는 것을 추천한다.

Comparator를 사용하여 s1과 s2를 비교하려고 한다. 이때 s1, s2는 아직 가공이 되지 않는 files 배열의 값들 중 하나이다. 때문에 가공을 위해 detach 메서드를 호출하여 file1, file2 배열에 각각 저장해준다.
맨 처음 head를 비교하면 값이 나온다. 그러나 이때 0이 나오게 된다면 두 값이 같다는 뜻이다. head가 같다면 number를 통해서 비교를 해주어야 하므로 number가 담겨있는 file1[1]과 file2[1]을 int형으로 바꿔준 뒤에 뺄셈 연산을 통해 우선순위를 비교해준다.

이처럼 모든 정렬이 끝이 난 뒤 files 배열을 반환해주면 문제를 해결할 수 있다.

---

💡느낀 점

데이터의 가공은 여러 번 해봤기 때문에 생각보다 간단하게 해결할 수 있었다. 그러나 정렬을 하는 부분에서 꽤 오랜 시간이 걸렸다. 특히 이번 문제를 풀면서 Comparable과 Comparator에 대해 다시 찾아보게 되었다. 문제는 어려웠지만 헷갈릴 수 있는 개념을 다시 한 번 확인할 수 있는 기회가 생겼다는 것에 만족했다.

---

링크

문제 링크