2018 KAKAO BLIND RECRUITMENT : [3차] 파일명 정렬

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

코딩테스트 연습 - [3차] 파일명 정렬

 

// 코드

import java.util.*;
 
class Solution {
    
    
    // filename, heade, number 정보를 관리하는 클래스
    class File {
        String fileName;
        String head;
        int number;
        
        public File(String fileName, String head, int number) {
            this.fileName = fileName;
            this.head     = head;
            this.number   = number;
        }        
    }
    
    
    // Head와 Number로 분리한다.
    String[] splitFileName(String file) {
        
        String[] str = new String[3];
        str[0] = "";
        str[1] = "";
        str[2] = "";
        
        int idx = 0;
        for(int i=0; i<file.length(); ++i) {
            
            char c = file.charAt(i);    
            // 문자인 경우 
            if( idx == 0 ) {
                if(!Character.isDigit(c)) str[idx] += c;
                else ++idx;
            }  
            // 숫자인 경우
            if( idx == 1 ) {
                if(Character.isDigit(c)) str[idx] += c;
                else ++idx;
            }    
            // 마지막 꼬리인 경우
            if( idx >= 2 ) str[idx] += c;            
        }
        
        // 문자열은 대/소문자를 구분하지 않으므로 소문자로 치환해서 반환한다.
        str[0] = str[0].toLowerCase();
        return str;                
    }
    
    
    // main
    public String[] solution(String[] files) {
        
        // 파일객체들을 담는 배열
        File[] fileInfoArr = new File[files.length];
        
        for(int i=0; i<files.length; ++i) {
            String[] str   = splitFileName(files[i]); // 이름을 분리한다.
            fileInfoArr[i] = new File(files[i], str[0], Integer.parseInt(str[1])); // 파일배열에 파일 객체를 넣어준다.
        }
        
        // 문제에서 제시한 조건에 맞도록 정렬한다.
        Arrays.sort(fileInfoArr, new Comparator<File>() {           
            public int compare(File o1, File o2) {
                if( (o1.head).equals(o2.head) ) return o1.number - o2.number;
                else                            return (o1.head).compareTo(o2.head); 
            }            
        });
        
        // 정렬된 순을 answer 배열에 담아주고 반환한다.
        String[] answer = new String[files.length];
        for(int i=0; i<files.length; ++i) answer[i] = fileInfoArr[i].fileName;
        return answer;
    }
    
    
}