[프로그래머스] 파일명 정렬

문제

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

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

문제해결방법

각 파일명을 입력받아 HEAD, NUMBER와 들어온 순서를 기록하고 각 파일의 해당 값들을 비교하며 정렬해준다.

코드

package Programmers;
 
import java.util.Arrays;
import java.util.PriorityQueue;
 
class File implements Comparable<File> {
    String fileName;
    String head;
    int number;
    int originNum;
    
    File(String fileName, int index) {
        this.fileName = fileName;
        this.originNum = index;
        int s = 0, e = 0;
        for(int i=0; i<fileName.length(); i++) {
            if(fileName.charAt(i)>='0' && fileName.charAt(i)<='9') {
                e = i;
                break;
            }
        }
        this.head = fileName.substring(s, e).toLowerCase();
        s = e;
        for(int i=s; i<fileName.length(); i++) {
            if(fileName.charAt(i)<'0' || fileName.charAt(i)>'9') {
                e = i;
                break;
            }
        }
        if(s==e) e = fileName.length();
        this.number = Integer.parseInt(fileName.substring(s, e));
    }
    
    @Override
    public int compareTo(File arg0) {
        if(arg0.head.equals(this.head)) {
            if(arg0.number==this.number) return Integer.compare(this.originNum, arg0.originNum);
            else return Integer.compare(this.number, arg0.number);
        }
        return this.head.compareTo(arg0.head);
    }
}
 
public class Solution_파일명정렬 {
 
    public static String[] solution(String[] files) {
        PriorityQueue<File> pq = new PriorityQueue<File>();
        for(int i=0; i<files.length; i++) pq.add(new File(files[i], i));
        String[] answer = new String[pq.size()];
        for(int i=0; i<answer.length; i++) answer[i] = pq.poll().fileName;
        return answer;
    }
    
    public static void main(String[] args) {
        String[] files = {"F-5 Freedom Fighter", "B-50 Superfortress", "A-10 Thunderbolt II", "F-14 Tomcat"};
        System.out.println(Arrays.toString(solution(files)));
    }
}

코드에 대한 피드백이나 더 좋은 아이디어는 언제나 환영입니다.