문자열 내 마음대로 정렬하기

https://school.programmers.co.kr/learn/courses/30/lessons/12915

프로그래머스

Comparator를 sort 내부에 정의할 수 있는걸 알게 되었다, n번째 char가 다르면 기존에 comparator 사용하는 것처럼 하지만 같으면 string1.compareTo(string2) 형태로 비교하면 사전상 앞에 있는 걸로 비교할 수 있다

import java.util.*;

public class Solution {
    public String[] solution(String[] strings, int n) {
        Arrays.sort(strings, new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                if (o1.charAt(n) == o2.charAt(n)) return o1.compareTo(o2);
                else if (o1.charAt(n) > o2.charAt(n))
                    return  1;
                else return  -1;
            }
        });
        return strings;
    }

    public static void main(String[] args) {
        Solution T = new Solution();
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        String[] arr = new String[n];
        for (int i = 0; i < n; i++) arr[i] = in.next();
        int m = in.nextInt();
        for(String x : T.solution(arr,m));
    }
}