<문자열 내 마음대로 정렬하기> 풀이

문제 설명

문자열로 구성된 리스트 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"]

---

문제풀이

첫 시도

문제의 요구사항인 string의 특정 index 문자를 추출해서 비교하기 위해 String 클래스 내부의 charAt(int index) 메소드를 사용하려는 계획을 세웠다. 각각의 원소인 string 의 charAt()을 비교하고 순서를 바꿔주어 큰 범위에서 정렬을 할 수 있었지만, 세부적인 부분에서 문제가 발생했다. 예를 들면, "abcd" 와 "abc"를 비교하여 abc를 abcd앞으로 바꿔주는 사전식 정렬을 해줘야했는데 여기서 머리가 멈췄다.

어떤식으로 정렬을 해야할지 한참을 고민하다가 결국 다른사람의 풀이를 봤는데, 기발한 아이디어에 감탄이 나왔다.

 

기발한 아이디어

핵심 아이디어를 간략하게 단계별로 정리하면 다음과 같다.

Step1 : <Key Idea> 비교의 기준이 되는 문자를 맨 앞에 추가한 새로운 String을 원소로 가지는 배열을 만든다.
Step2 : 라이브러리를 이용하여 새로운 배열을 정렬한다.
Step3 : 정렬한 배열에서 기준이 되는 문자를 제외한 나머지를 추출한다.

 

위의 과정을 토대로 코드를 구성하면 굉장히 간결한 결과물을 얻을 수 있다.

import java.util.ArrayList;
import java.util.Collections;

class Solution {
    public String[] solution(String[] strings, int n) {
        String[] answer = new String[strings.length];
        ArrayList<String> arr = new ArrayList<>();
        for(String str : strings){ // 기준이 되는 문자를 맨 앞에 추가하는 과정
            arr.add(str.charAt(n) + str); //문자열에서 + 는 이어붙이는 연산
        }
        Collections.sort(arr); // 정렬(기본적으로 오름차순)
        for(int i =0; i < arr.size(); i++){
            answer[i] = arr.get(i).substring(1);//맨 앞 글자 제외하고 나머지 할당
        }
        return answer;
    }
}

오늘의 교훈

역시 가장 중요한 것은 문제에 대한 접근법이다.