PROGRAMMERES Level 1 연습문제 문자열 내 마음대로 정렬하기 배열 (JAVA 자바)

import java.util.*;

class Solution {
    public String[] solution(String[] strings, int n) {

        // n번째 인덱스값을 저장할 ArrayList
        ArrayList<String> arr = new ArrayList<>();
        
        // strings 배열의 i번째 인덱스의 n번째 인덱스값을 charAt으로 뽑아주고 strings 배열의 i번째 인덱스값 
        // 을 더해서 arraylist에 넣어줌. ex("usun", "ebed", "acar") -> 정렬을 이용하기위해서
        for(int i=0; i<strings.length; i++){
            arr.add(strings[i].charAt(n) + strings[i]);            
        }
        
        // 정렬
        Collections.sort(arr);
        
        // 배열 초기화
        String answer[] = new String[arr.size()];
        
        // 정렬을 하기 위해 더해준 n번째 인덱스값을 짜르고 배열에 저장
        for(int i=0; i<strings.length; i++){
            answer[i] = arr.get(i).substring(1);           
        }

        return answer;
    }
}