🔥[99클럽 코테 스터디] 3일차 TIL - 문자열 내 마음대로 정렬하기

⏳문제

문제 설명

문자열로 구성된 리스트 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) {
        String[] answer = new String[strings.length];
       
        strings = SWAP(strings, n);
        Arrays.sort(strings);
        strings = SWAP(strings, n);
        
        answer = strings;
        return answer;
    }
    
    public String[] SWAP(String[] strings, int n) {
        for (int i = 0; i < strings.length; i++) {
            String[] value = strings[i].split("");
            String tmp = value[0];
            value[0] = value[n];
            value[n] = tmp;
            
            strings[i] = String.join("", value);
        }
        return strings;    
    }
}

나는 처음에 n번째 인덱스의 문자를 기준으로 정렬하기 위해서 해당 문자의 인덱스와 첫 번째 인덱스의 문자를 바꾼 후 정렬 함수를 사용하면 될 것이라 생각했다.
그래서 무작정 문자열을 split()함수로 나누고, n번째 문자를 꺼내는 과정과, 꺼낸 문자와 맨 앞 요소와 자리를 바꿔준 후 다시 문자들을 join()함수를 통해 재조립 시키는 SWAP함수라는 것을 만들어 strings를 입력값으로 넣었다.
그러면 n번째 인덱스의 문자와 첫 번째 인덱스의 문자가 서로 바뀌게 되고, 이 상태에서 Arrays.sort()함수를 사용해 오름차순 정렬을 해주면 된다고 생각했다.

마지막으로, 바뀐 문자 순서를 복구시키기 위해 SWAP함수를 다시 호출함으로써 문제를 해결했다고 생각이 드는 찰나..

통과한 케이스도 있었지만, 대부분의 테스트 케이스에서 실패가 떴다.

실패 원인

알고보니 제한 조건 중 마지막 조건을 간과한 것이었다!

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

기존의 방법대로 라면, n번째 인덱스에 위치한 문자가 서로 같은 경우와 상관 없이 첫 번째 인덱스와 위치를 바꾼 상태에서 정렬을 하였는데, 그렇게 되면 정렬의 결과가 달라질 수 있다는 점을 생각하지 못한 것이다.! 그래서 다른 방법을 찾아보기로 했다.

🖥️최종 코드

import java.util.*;

class Solution {
    public String[] solution(String[] strings, int n) {
        Arrays.sort(strings, (a, b) -> {
           if (a.charAt(n) != b.charAt(n)) {
               return Character.compare(a.charAt(n), b.charAt(n));
            }
            else {
                return a.compareTo(b);
            }
        });
        
        return strings;
    }
}

다른 방법을 고민하고 다른 사람의 풀이를 찾아보다가 Arrays.sort()의 두 번째 인자로 Comparator인터페이스를 람다식으로 구현하는 방법을 찾았다.

이 람다 식은 정렬 기준을 정의하는 역할을 한다.
즉, 내가 구하고자 하는 정렬 기준을 여기다가 정의하면 되는 것이다!

그래서 정렬 기준으로 두 가지 경우를 나누어 주었다.

• n번째 문자가 다를 경우
• n번째 문자가 같을 경우

n번째 문자가 다를 경우에는 해당 인덱스의 문자를 charAt(n)함수를 통해 구한 다음 Character.compare()함수에 인자로 넣어주었다. 여기서 Character.compare()함수는 두 개의 문자를 비교할 때 사용하는 함수이다.
그러므로 a가 b보다 크면 양수 값을 반환하고, 작으면 음수 값을 반환하게 됨으로써 두 문자열을 오름차순으로 정렬할 수 있다.

n번째 문자가 같을 경우에는 두 문자열을 compareTo()함수를 사용하여 정렬 순서를 반환해주면 된다. Character.compare()함수는 문자를 비교하지만, compareTo()함수는 문자열을 비교하는 함수이다. 단, 비교에 의한 반환값은 서로 동일하다.

이렇게 해서 모든 테스트 케이스를 통과할 수 있었다!

---

🔗문제 링크
💻Repository