[프로그래머스 Level.2] [3차] 파일명 정렬 (2018 KAKAO BLIND RECRUITMENT) (Java)

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/17686

프로그래머스

 

 

 

코딩테스트 연습 > 2018 KAKAO BLIND RECRUITMENT > [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"]

 

 

 

 

나의 코드

import java.util.*;

class Solution {
    public String[] solution(String[] files) {
        String[] answer = {};
        
        Arrays.sort(files, new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                String[] file1 = getFile(o1);
                String[] file2 = getFile(o2);
                
                if(file1[0].equals(file2[0])) // head가 같은 경우 number 기준 오름차순
                   return Integer.parseInt(file1[1]) - Integer.parseInt(file2[1]);
                
                return file1[0].compareTo(file2[0]); // head가 다른 경우 head 기준 오름차순
            }
        });
        
        answer = files;
        
        return answer;
    }
    
    public String[] getFile(String fileName) {
        String head = "";
        String number = "";
        String tail = "";
        int index = 0;
        
        for(int i=index; i<fileName.length(); i++) {
            if(fileName.charAt(i) >= '0' && fileName.charAt(i) <= '9') {
                head = fileName.substring(0, i);
                index = i;
                break;
            }
        }
        
        for(int i=index; i<fileName.length(); i++) {
            if(fileName.charAt(i) < '0' || fileName.charAt(i) > '9') {
                number = fileName.substring(index, i);
                tail = fileName.substring(i);
                break;
            }
            
            // TAIL이 아무 글자도 없는 경우
            if(i == fileName.length()-1 && fileName.charAt(i) >= '0' && fileName.charAt(i) <= '9') {
                number = fileName.substring(index);
                break;
            }
        }
        
        return new String[] {head.toUpperCase(), number, tail};
    }
}

 

풀이

1. 이 문제는 files 배열을 문제에 주어진대로 정렬하여 리턴하면 되는 문제이다.
2. files 배열을 sort해야 하는데 files의 원소를 String 형인 head, number, tail로 자른 후에 head끼리 비교하여 head로 오름차순하고 head가 같은 경우에는 number를 int형으로 변환하여 비교하면 된다.
3. 파일명을 head, number, tail로 자르는게 중요한데 인덱스를 저장하며 파일명을 자르기 위해 int형으로 index 변수를 생성한다.
4. head는 숫자가 없으므로 파일명 크기만큼 첫 인덱스부터 for문을 돌다가 숫자가 나오면 해당 인덱스 전까지 문자를 substring()을 통해 문자열을 잘라 head에 저장하고 number를 찾을 때 해당 인덱스부터 시작 하기 위해서 해당 인덱스를 index 변수에 저장한다.
5. 그리고 index 부터 나머지 파일명 부분을 for문을 돌면서 number와 tail을 찾으면 된다.
   number는 숫자만 가능하므로 for문을 돌다가 숫자가 아닌 수가 나오면 index부터 해당 인덱스 전까지 문자를 substring()을 통해 잘라 number에 저장하고 나머지 뒤의 부분을 tail에 저장하면 된다.
   문제에 주어졌듯이 tail이 없을 수도 있기 때문에 for문을 돌면서 마지막 인덱스까지 진행했는데도 숫자만 나왔다면 index부터 끝까지를 number에 저장하면 된다.
6. 이렇게 head, number, tail을 String 배열로 담아 반환하면 되는데 파일명을 정렬할 때 head 부분의 대소문자를 구분하지 않는다고 했으므로 head는 toUpperCase()를 이용하여 대문자로 변환한다.
7. 이렇게 파일명을 head, number, tail로 나누어 String 배열로 담아 전달하고 이것을 이용하여 정렬하여 리턴하면 된다.