유블로그

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

알고리즘

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

yujeong kang 2021. 2. 16. 01:54

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

 

소요시간 : 24분

 

중간중간 실수를 잡느라 문제 난이도에 비해 오래 걸렸다 ,, ㅎㅎ

문제만 길었지 매우 쉬운 문제..

 

head를 숫자나오기 전 인덱스까지 잘라서 대소문자 구분없이 정렬한다.

다음 number 부분을 1~5자리까지 잘라서 integer 순으로 정렬한다.

끝!

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

class Solution {
  public String[] solution(String[] files) {
    Arrays.sort(files, new Comparator<String>() {
      @Override
      public int compare(String o1, String o2) {
        int idx1 = -1, idx2 = -1;
        for (int i = 0; i < o1.length(); i++) {
          if('0' <= o1.charAt(i) && o1.charAt(i) <= '9') {
            idx1 = i;
            break;
          }
        }
        for (int i = 0; i < o2.length(); i++) {
          if('0' <= o2.charAt(i) && o2.charAt(i) <= '9') {
            idx2 = i;
            break;
          }
        }
        String tmp1 = o1.substring(0, idx1);
        String tmp2 = o2.substring(0, idx2);

        if(!tmp1.equalsIgnoreCase(tmp2))
        	return tmp1.compareToIgnoreCase(tmp2);

        int cnt1 = 0;
        for (int i = idx1; i < idx1+5; i++) {
          if(i == o1.length()) break;
          if(cnt1 == 5) break;
          if('0' <= o1.charAt(i) && o1.charAt(i) <= '9') {
          	cnt1++;
          }
          else break;
        }
        int cnt2 = 0;
        for (int i = idx2; i < idx2+5; i++) {
          if(i == o2.length()) break;
          if(cnt2 == 5) break;
          if('0' <= o2.charAt(i) && o2.charAt(i) <= '9') {
          	cnt2++;
          }
          else break;
        }

        return 
          Integer.parseInt(o1.substring(idx1, idx1+cnt1))
          - Integer.parseInt(o2.substring(idx2, idx2+cnt2));
      }
    });

    return files;
  }
}