[프로그래머스] 17686번 - 파일명 정렬 (Java)

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

프로그래머스

---

문제

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

저장소 서버에는 프로그램의 과거 버전을 모두 담고 있어, 이름 순으로 정렬된 파일 목록은 보기가 불편했다. 파일을 이름 순으로 정렬하면 나중에 만들어진 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 부분을 기준으로 사전 순으로 정렬한다. 이때, 문자열 비교 시 대소문자 구분을 하지 않는다. 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는 함께 입력으로 주어질 수 있다.)

 

출력 형식

위 기준에 따라 정렬된 배열을 출력한다.

---

풀이

문제에서 구하고자 하는 것은 파일명 배열이 들어오면 그것을 head와 number를 기준으로 정렬하는 것입니다. 포인트는 number가 문자열로 정렬되는 것이 아니라 숫자일 때의 오름차순 정렬을 해야 합니다.

 

head와 number, tail로 정렬을 해야 하기 때문에 각각을 먼저 구해야 합니다.

head와 number와 tail을 나누는 기준은 다음과 같습니다.

 

1) head는 파일명의 처음부터 숫자가 나오기 전까지의 문자열

2) number는 처음 숫자가 나온 부분부터 최대 다섯 자리까지의 숫자

3) tail은 number 이후의 모든 문자열

 

만약 파일명이 "file123456.png"로 들어온다면 head는 "file", number는 12345, tail은 "6.png"가 될 것입니다.

사실 채점 테스트 케이스에 number가 최대 다섯자리라는 조건에 해당하는 케이스가 없는건지 최대 다섯자리라는 조건은 적용하지 않아도 통과는 하지만, 그래도 문제의 조건에 맞게 풀기 위해서 최대 다섯자리까지만 number로 했습니다!

 

입력으로 들어오는 files 배열의 문자열들을 한 글자씩 읽으면서 head, number, tail을 구해주면 됩니다.

각 글자가 어느 부분에 해당하는지 구별하기 위해 boolean 변수 isHead, isNumber, isTail을 만들어 줬습니다.

 

1. head 구하기

head는 파일명의 첫 글자부터 시작이기 때문에 isHead는 true로 초기화 하고 시작합니다.

숫자가 나오면 그 전 글자까지가 head 이므로 지금 읽고 있는 글자가 숫자라면 isHead는 false가 됩니다.

 

2. number 구하기

number는 head 다음에 나오는 숫자 최대 다섯자리입니다. 어떤 글자가 숫자라면 그 글자부터 isNumber는 true가 됩니다. 그 이후에 숫자가 아닌 문자가 나왔다면, 혹은 number의 길이가 이미 5 이상이라면 isNumber는 false가 되고 isTail은 true가 됩니다.

head와 number를 구할 때는 이 문자가 숫자인지 아닌지를 구분해야 하는데 Integer.parseInt()와 NumberFormatException 에러를 이용해서 try catch 문으로 구현하였습니다.

 

3. tail 구하기

tail은 number 이후에 나오는 문자열 전체이기 때문에 isTail이 true가 되는 순간부터 문자열의 끝까지가 tail이 됩니다.

String head = "";
String number = "";
String tail = "";

boolean isHead = true;
boolean isNumber = false;
boolean isTail = false;

for(int i=0; i<file.length(); i++){
    // head, number, tail 찾기
    // number의 길이는 최대 5 **********
    if(!isTail && isInteger(String.valueOf(file.charAt(i)))){
        isHead = false;
        isNumber = true;
    }

    if((isNumber && !isInteger(String.valueOf(file.charAt(i)))) || number.length() >= 5){
        isNumber = false;
        isTail = true;
    }

    if(isHead)
        head += String.valueOf(file.charAt(i));

    if(isNumber)
        number += String.valueOf(file.charAt(i));

    if(isTail){
        tail += String.valueOf(file.charAt(i));
    }

}

 

head, number, tail을 모두 구했다면 head와 number를 기준으로 정렬을 해야 합니다.

구한 세 부분을 순서대로 2차원 배열에 저장하고 그 2차원 배열을 조건에 맞게 정렬을 해주면 됩니다.

 

먼저 head를 대소문자 구분 없이 사전식으로 정렬해주고, head가 같다면 number를 오름차순으로 정렬하면 됩니다.

Arrays.sort(file_names, new Comparator<String[]>(){
    @Override
    public int compare(String[] o1, String[] o2){
        if(o1[0].toUpperCase().equals(o2[0].toUpperCase())){
            return Integer.parseInt(o1[1]) - Integer.parseInt(o2[1]);
        }

        return o1[0].toUpperCase().compareTo(o2[0].toUpperCase());
    }
});

 

그러면 2차원 배열이 조건에 맞게 정렬이 되어 있으므로 배열에 저장된 head, number, tail을 붙여 answer 배열에 넣어주면 됩니다.

 

---

소스코드

import java.util.*;

class Solution {
    public String[] solution(String[] files) {
        String[] answer = new String[files.length];
        
        String[][] file_names = new String[files.length][3];
        
        for(int k=0; k<files.length; k++){
            String file = files[k];
            
            String head = "";
            String number = "";
            String tail = "";
            
            boolean isHead = true;
            boolean isNumber = false;
            boolean isTail = false;
            
            for(int i=0; i<file.length(); i++){
                // head, number, tail 찾기
                if(!isTail && isInteger(String.valueOf(file.charAt(i)))){
                    isHead = false;
                    isNumber = true;
                }
                
                if((isNumber && !isInteger(String.valueOf(file.charAt(i)))) || number.length() >= 5){
                    isNumber = false;
                    isTail = true;
                }
                
                if(isHead)
                    head += String.valueOf(file.charAt(i));
                
                if(isNumber)
                    number += String.valueOf(file.charAt(i));
                
                if(isTail){
                    tail += String.valueOf(file.charAt(i));
                }
                
            }
            
            file_names[k][0] = head;
            file_names[k][1] = number;
            file_names[k][2] = tail;
            
        }
        
        // 정렬
        Arrays.sort(file_names, new Comparator<String[]>(){
           @Override
            public int compare(String[] o1, String[] o2){
                if(o1[0].toUpperCase().equals(o2[0].toUpperCase())){
                    return Integer.parseInt(o1[1]) - Integer.parseInt(o2[1]);
                }

                return o1[0].toUpperCase().compareTo(o2[0].toUpperCase());
            }
        });
        
        for(int i=0; i<answer.length; i++){
            answer[i] = file_names[i][0] + file_names[i][1] + file_names[i][2];
        }
        
        return answer;
    }
    
    public boolean isInteger(String str){
        try{
            Integer.parseInt(str);
            return true;
        }
        catch(NumberFormatException ex){
            return false;
        }
    }
}