[JAVA/프로그래머스] 문자열 내 마음대로 정렬하기

 

문제 개요

주어진 문자열 배열 'strings'  와 정수 'n'에 대해, 배열 내 각 문자열의 'n'번째 문자를 기준으로 오름차순 정렬하는 문제입니다. 이 문제의 핵심은 특정 조건에 따라 문자열을 정렬하는 맞춤 정렬 로직을 구현 하는 것 입니다.

 

코드

import java.util.Arrays;
import java.util.Comparator;

class Solution {
    public String[] solution(String[] strings, int n) {
        // Comparator를 이용한 정렬
        Arrays.sort(strings, new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                // n번째 문자를 기준으로 비교
                if(o1.charAt(n) > o2.charAt(n)) {
                    return 1;
                } else if(o1.charAt(n) < o2.charAt(n)) {
                    return -1;
                } else {
                    // n번째 문자가 동일할 경우 전체 문자열을 사전순으로 비교
                    return o1.compareTo(o2);
                }
            }
        });
        return strings;
    }
}

• 'Comparator' 인터페이스는 'compare' 메소드를 오버라이드하여 두 문자열 ('o1','o2') 의 'n'번째 문자를 비교하여 정렬 순서를 결정합니다
• 두 문자의 ASCII 값 비교를 통해 두 문자가 같은지, 어느하나가 다른지 결정을 하며 같은 결우 두 문자열을 전체를 사전순으로 비교합니다.

성능

• 시간 복잡도: 이 알고리즘은 'Arrays.sort'함수의 시간 복잡도인 O(n log n) 를 따릅니다. 여기서 n은 배열 'strings'의 길이입니다.