[level 2] [3차] 파일명 정렬 - 17686

문제 링크

프로그래머스

import java.util.Arrays;

class Solution {

    public String[] solution(String[] files) {
       Arrays.sort(files, (str1, str2) -> {
             String head1 = parseHead(str1.toLowerCase());
             String head2 = parseHead(str2.toLowerCase());
             if (head1.compareTo(head2) == 0) {
                Integer number1 = parseNumber(str1.substring(head1.length()));
                Integer number2 = parseNumber(str2.substring(head2.length()));
                return number1.compareTo(number2);
             }
             return head1.compareTo(head2);
          });
       return files;
    }

    private String parseHead(String str) {
       StringBuilder stringBuilder = new StringBuilder();
       for (char c : str.toCharArray()) {
          if (Character.isDigit(c)) {
             break;
          }
          stringBuilder.append(c);
       }
       return stringBuilder.toString();
    }

    private Integer parseNumber(String str) {
       StringBuilder stringBuilder = new StringBuilder();
       for (char c : str.toCharArray()) {
          if (!Character.isDigit(c) || stringBuilder.length() == 5) {
             break;
          }
          stringBuilder.append(c);
       }
       return Integer.valueOf(stringBuilder.toString());
    }
}