프로그래머스 - 문자열 내 마음대로 정렬하기 (JAVA)

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

프로그래머스

 

문제)

배열의 각 원소(string)를 n번째 글자 기준으로 오름차순 정렬한다.

만약 n번째 글자가 같다면, 사전순으로 정렬한다.

 

접근 방식) 

1.  Arrays.sort()를 사용해서, 정렬 기준을 정의

    n번째 글자 기준으로 오름차순 정렬하되, s1과 s2의 n번째 글자가 같다면, 사전순으로 정렬한다.

2. 정렬된 strings 배열을 return 한다.

 

import java.util.Arrays;
class Solution {
    public String[] solution(String[] strings, int n) {
        
        Arrays.sort(strings, (s1, s2) -> {
            if (s1.charAt(n) == s2.charAt(n))
                return s1.compareTo(s2);
            return s1.charAt(n) - s2.charAt(n);
        });
        
        return strings;
    }
}