[알고리즘 PS(프로그래머스)] 문자열 내 마음대로 정렬하기 Java 문제 풀이

 

해당 문제는 Level01 문제이다.

문제를 해결하는데 두 가지의 이해가 필요하다.

1. Sort
2. Comparator 혹은 Comparable

접근

문제는 주어진 n번에 해당하는 인덱스를 기준으로 졍렬하는 것이다. 하지만

문제에 주어진 조건을 보면

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

이 마지막 조건이 우리에게 구체적으로 정렬할 기준을 줬다.

정렬 기준

1. n번째 인덱스의 사전순
2. n번째 인덱스가 같을 경우 전체 문자열 사전순

원래 Arrays.sort나 Collections.sort의 정렬 기준은 문자열의 사전순으로 기본 정의되어 있기 때문에 정렬 기준을 바꿔야 한다.

정렬 기준 변경

정렬 기준은 Comparator의 메서드를 재정의 하는 것에서 부터 시작된다. 만약 Comparator로 정렬 기준 변경하기 나 Comparable로 정렬 기준 변경하기에 대해서 모른다면 해당 포스팅에서 확인하고 오는 것도 좋은 선택이다.

1번 정렬 기준
1번 정렬 기준은 구체적으로 compare 클래스를 재정의 해서 기준을 바꿔야 한다.

Arrays.sort(strings, new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                if(o1.charAt(n) > o2.charAt(n)){
                    return 1;
                } else {
                    return -1;
                }
            }
        });

2번 정렬 기준
2번 정렬 기준은 기본 정렬 기준으로 적용해야하기 때문에 기본 정렬을 수행한다.

Arrays.sort(strings, new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                if(o1.charAt(n) > o2.charAt(n)){
                    return 1;
                } else if(o1.charAt(n) == o2.charAt(n)){
                    //2번 기준
                    return o1.compareTo(o2); // 기본 문자열 사전순 정렬
                }
                else {
                    return -1;
                }
            }
        });

을 이용하면