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

 

문제 설명 : 

문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 [sun, bed, car]이고 n이 1이면 각 단어의 인덱스 1의 문자 u, e, a로 strings를 정렬합니다.

 

제한 조건 : 

• strings는 길이 1 이상, 50이하인 배열입니다.
• strings의 원소는 소문자 알파벳으로 이루어져 있습니다.
• strings의 원소는 길이 1 이상, 100이하인 문자열입니다.
• 모든 strings의 원소의 길이는 n보다 큽니다.
• 인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

입출력 예 : 

strings	n	return
[sun, bed, car]	1	[car, bed, sun]
[abce, abcd, cdx]	2	[abcd, abce, cdx]

 

입출력 예 설명 : 

입출력 예 1
sun, bed, car의 1번째 인덱스 값은 각각 u, e, a 입니다. 이를 기준으로 strings를 정렬하면 [car, bed, sun] 입니다.

입출력 예 2
abce와 abcd, cdx의 2번째 인덱스 값은 c, c, x입니다. 따라서 정렬 후에는 cdx가 가장 뒤에 위치합니다. abce와 abcd는 사전순으로 정렬하면 abcd가 우선하므로, 답은 [abcd, abce, cdx] 입니다.

 

 

import java.util.*;

class Solution {
    public String[] solution(String[] strings, int n) {
        Arrays.sort(strings, new Comparator<String>() {
          @Override
          public int compare(String s1, String s2){
              char c1 = s1.charAt(n);
              char c2 = s2.charAt(n);
              
              // n번째 문자가 같을 경우에 원본 스트링을 사전순으로
              if(c1 == c2){
                  return s1.compareTo(s2);
              } else return c1 - c2;
          }
      });
        return strings;
    }
}

 

알고리즘 풀이 : 

1. Arrays.sort를 사용하여 문자열 배열 strings를 Comparator을 사용하여 정렬

2. c1과 c2가 같을 경우에는 s1.compareTo(s2)를 return을 사용해 반환

- 인덱스 문자가 같을 경우에는 사전 순으로 더 빠른 문자를 앞으로 오도록 정렬

3. c1과 c2가 다를 경우에는 c1 - c2를 리턴

- n이 값이 작은 문자열로 정렬

 

 

Comparator ?

- 기본적으로 Object를 매개 변수로 받아 비교

- 정렬 대상이 primitive 타입이면 wrapper 클래스를 이용해 Object로 만들어 매개 변수로 넘겨야 함

 

 

KM Comment :

문제에서 자주 나오는 정렬이 아니라 바로 Comparator을 생각해 내지 못했다.

다른 풀이도 많지만 이번 Comparator 익명 클래스를 처음 접하여 의미와 사용에 대해 배울 수 있는 문제였다.

비슷한 여러 문제를 접해봐야 손에 익을 것 같은 코드라 추가 공부가 필요하다고 생각한다.