[Programmers] 문자열 내 마음대로 정렬하기 - Java

문제

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

프로그래머스

 

풀이

문자열 정렬 기준(Arrays.sort()에 있는 람다 구현)을 생각하면 되는 문제이다.

1. n번째 있는 인덱스의 문자와 비교해서 같으면 문자열 오름차순 정렬한다.

2. n번째 있는 인덱스의 문자와 비교해서 다르면 문자로 오름차순 정렬한다.

3. 정렬한 배열을 반환한다.

 

소스코드

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;
    }
}

 

실행결과