문자열 내 마음대로 정렬하기 // Arrays.sort(), Comparator<T>

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

프로그래머스

 

 

오답 : 

import java.util.ArrayList;

class Solution {
    public String[] solution(String[] strings, int n) {
        ArrayList<String> arrayList = new ArrayList<>();
        for (String str : strings) {
            arrayList.add(str);
        }

        for (int i = 0; i < arrayList.size() - 1; i++) {
            for (int j = i + 1; j < arrayList.size(); j++) {
                char charAtI = arrayList.get(i).charAt(n);
                char charAtJ = arrayList.get(j).charAt(n);

                if (charAtI > charAtJ) {
                    String temp = arrayList.get(i);
                    arrayList.set(i, arrayList.get(j));
                    arrayList.set(j, temp);
                } 
            }
        }
        String[] answer = arrayList.toArray(new String[arrayList.size()]);
        
        return answer;
    }
}

 

이는 문제의

• 인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

조건을 충족하지 못합니다.

 

 

정답: 

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

class Solution {
    public String[] solution(String[] strings, int n) {
        Arrays.sort(strings, new Comparator<String>() {
            @Override
            public int compare(String s1, String s2) {
                // n번째 글자 비교
                if (s1.charAt(n) == s2.charAt(n)) {
                    // n번째 글자가 같으면 사전순으로 정렬
                    return s1.compareTo(s2);
                }
                // n번째 글자가 다르면 해당 글자 비교
                return Character.compare(s1.charAt(n), s2.charAt(n));
            }
        });

        return strings;
    }
}

 

 

Arrays.sort() 에 의해 strings 를 정렬하는데, 그 기준은 new Comparator<String>() {} 입니다.

 

 

Comparator<T> : 두 객체를 비교하기 위한 인터페이스입니다. 객체를 비교하는 로직을 정의하여 정렬, 탐색 등의 작업에서 사용할 수 있습니다. Comparator에는 int compare() , boolean equals(), default Comparator<T> reversed() 등 다양한 메서드가 있습니다.

 

그 중 int compare()를 오버라이딩하여 메서드를 재정의해 사용했습니다.

 

 

 

this.compareTo(T object) :  두 객체를 비교합니다.

• 반환값이 음수이면 this가 object 보다 작다는 뜻입니다.
• 반환값이 0이면 같다는 의미입니다.
• 반환값이 양수이면 this가 object 보다 크다는 뜻입니다.

 

s1.compareTo(s2) : 

 

• compareTo()는 문자열의 유니코드 값을 기반으로 문자열을 사전순으로 비교합니다.

 

 

 

Character.compare(o1, o2) : 두 값을 비교할 때 사용됩니다.

 

• 반환값이 음수이면 o1이 o2보다 작다는 뜻입니다.
• 반환값이 0이면 두 객체가 같다는 의미입니다.
• 반환값이 양수이면 o1이 o2보다 크다는 뜻입니다.

 

Character.compare(s1.charAt(n), s2.charAt(n)) : 

 

• 만약 s1과 s2의 n번째 문자가 다르면, 이 문자를 비교합니다.
• Character.compare(s1.charAt(n), s2.charAt(n))는 두 문자의 유니코드 값을 비교해, 작은 값이 먼저 오도록 정렬합니다.
  • 유니코드 값이 작은 문자가 앞에 와야 하므로, 이 메서드는 n번째 문자를 기준으로 두 문자열을 정렬하는 역할을 합니다.

 

 

---

 

 

 

간단히 정리하자면 

 

new Comparator(T) {} : 를 오버라이딩하여 사용할 때

 

반환값이 음수가 나오게 하면 앞에 배치되고, 양수가 나오게 하면 뒤에 배치됩니다.

 

public class Main {
    public static void main(String[] args) {
        String[] arr = {"sunRise", "bed", "card"};
        Arrays.sort(arr, new Comparator<String>() {
            @Override
            public int compare(String s1, String s2) {
                return s1.length() - s2.length(); // 문자열 길이를 기준으로 정렬
            }
        });
        Arrays.stream(arr).forEach(System.out::println);
/*
bed
card
sunRise
*/

    }
}