[Java] programmers-"문자열 내 마음대로 정렬하기"

문제 설명

문자열로 구성된 리스트 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	result
["sun", "bed", "car"]	1	["car", "bed", "sun"]
["abce", "abcd", "cdx"]	2	["abcd", "abce", "cdx"]

---

알고리즘

1. strings 배열의 길이만큼 for문을 돌려서 해당 strings 앞에 n번째 문자를 붙여서 새로운 배열(sort)에 저장한다.
2. Arrays.sort() 함수를 사용해서 만들어진 sort 배열을 오름차순으로 정렬한다.
3. sort 배열의 첫번째 부분을 .substring()를 사용해서 없애준 후, answer에 저장한다.

---

내 소스 코드

import java.util.Arrays;
    
class Solution {
    public String[] solution(String[] strings, int n) {
        String[] answer = new String[strings.length];
        String[] sort = new String[strings.length];
        
        for(int i=0;i<strings.length;i++) {
            sort[i] = strings[i].charAt(n) + strings[i];
        }
        
        Arrays.sort(sort);
        
        for(int i=0;i<sort.length;i++) {
            answer[i] = sort[i].substring(1);
        }
        
        return answer;
    }
}