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

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

문제풀이

자바에서 정렬할 때는 항상 Comparable을 상속하여 사용하고는 했는데 이 문제에서와 같이 Comparator를 사용해야하는 경우도 있다. Comparator의 사용법을 기억하자!

• Comparator는 기본적으로 Object를 매개변수로 받아서 비교한다. 따라서, Comparator를 통해 정렬을 하고자할 때 정렬 대상이 primitive 타입이라면 wrapper 클래스를 이용해 Object로 만들어서 매개변수로 넘겨야 한다.

구현코드

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