[2018 카카오 블라인드 채용] 파일명 정렬

문제

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

코딩테스트 연습 - [3차] 파일명 정렬 | 프로그래머스

 

소스코드

import java.util.Arrays;

public class Solution {

    static class File implements Comparable<File> {
        String name, head;
        Integer number;

        public File(String name, String head, Integer number) {
            this.name = name;
            this.head = head;
            this.number = number;
        }

        @Override
        public int compareTo(File o) {
            return (this.head.equals(o.head))?
                    (this.number - o.number) :
                    (this.head.compareTo(o.head));
        }
    }

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

        return Arrays.stream(files)
                .map(Solution::makeFile)
                .sorted()
                .parallel()
                .map((file) -> file.name)
                .toArray(String[]::new);
    }

    public static File makeFile(String name) {

        String head = name.split("\\d")[0].toUpperCase();
        String number = Arrays.stream(name.split("\\D"))
                            .filter(s -> s.length()>0)
                            .findFirst().get();

        return new File(name, head, Integer.parseInt(number));
    }

}

 

회고

 문자열 문제에서 정규표현식이 유용하다는 걸 깨달았다. 내일 공부해야지...