[Programmers / Level 1] 12915. 문자열 내 마음대로 정렬하기(Java)

💡 Info

• 난이도: Lv1
  프로그래머스 링크: https://school.programmers.co.kr/learn/courses/30/lessons/12915
  풀이 링크(GitHub): hayannn/CodingTest_Java/프로그래머스/1/문자열 내 마음대로 정렬하기

입출력 조건

입출력 예시

문제 이해

• 각 문자열 앞에 n번째 문자를 붙여서 새롭게 정렬한 문자열 출력

알고리즘

풀이 시간 : 7분

1. 각 문자열 앞에 n번째 문자를 붙여서 -> 리스트로 저장
2. 리스트 정렬 -> n번째 문자 기준
3. 앞에 붙인 문자를 제거 -> 원래 문자열로 복원

import java.util.*;

class Solution {
    public String[] solution(String[] strings, int n) {
        return Arrays.stream(strings)
                .sorted((s1, s2) -> {
                    if (s1.charAt(n) == s2.charAt(n)) {
                        return s1.compareTo(s2);
                    }
                    return Character.compare(s1.charAt(n), s2.charAt(n));
                }).toArray(String[]::new);
    }
}

결과