[Programmers] 17686. [3차] 파일명 정렬

import java.util.Arrays;
import java.util.Comparator;

class Solution {

    public static String[] solution(String[] files) {


        File[] file_info = new File[files.length];

        for (int i = 0; i < files.length; i++) {
            String[] split = splitFileName(files[i]);
            file_info[i] = new File(files[i], split[0], Integer.parseInt(split[1]));
        }

        Arrays.sort(file_info, new Comparator<File>() {
            @Override
            public int compare(File o1, File o2) {
                if ((o1.head).equals(o2.head)) { 	// HEAD 부분이 같다면 
                    return o1.number-o2.number; 	// NUMBER 오름차순 정렬
                }else{
                    return (o1.head).compareTo(o2.head); // HEAD 부분을 기준으로 사전 순 정렬
                }
            }
        });

        String[] answer = new String[files.length];
        for (int i = 0; i < files.length; i++) {
            answer[i] = file_info[i].file_name;
        }

        return answer;
    }

    static class File {
        String file_name;
        String head;
        int number;

        public File(String file_name, String head, int number) {
            this.file_name = file_name;
            this.head = head;
            this.number = number;
        }
    }
    
    // file name을 head, number, tail로 나누는 함수
    public static String[] splitFileName(String file) {
        String str[] = new String[3];
        str[0] = ""; // HEAD
        str[1] = ""; // NUMBER
        str[2] = ""; // TAIL

        int idx = 0;
        for (int i = 0; i < file.length(); i++) {
            char c = file.charAt(i);

            if(idx==0 && !Character.isDigit(c)){ // 현재 처리 중인 부분이 HEAD이며, 현재 문자가 숫자가 아닌 경우
                str[idx] += c;
                continue;
            }

            if(idx==0 && Character.isDigit(c)){ // 현재 처리 중인 부분이 HEAD이며, 현재 문자가 숫자인 경우
                idx ++;
            }

            if(idx==1 && Character.isDigit(c)){ // 현재 처리 중인 부분이 NUMBER이며, 현재 문자가 숫자인 경우
                str[idx] += c;
                continue;
            }

            if(idx==1 && !Character.isDigit(c)){ // 현재 처리 중인 부분이 NUMBER이며, 현재 문자가 숫자가 아닌 경우
                idx ++;
            }
            str[idx] += c;
        }

        str[0] = str[0].toLowerCase(); // 대소문자 구분 X, 소문자로 변환

        return str;
    }
}