코딩 테스트

(Java) 프로그래머스 문자열 내 마음대로 정렬하기

로승리 2021. 11. 5. 02:30

리스트 맨 앞에 n번째 문자열을 붙이는게 핵심 포인트였다.

처음에는 문자열을 split올 다 잘라서 정렬하려고 했는데 생각처럼 잘 정렬이 되지 않았다.

검색을 통해 힌트를 얻어 해결했다.


최종 코드

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
class Solution {
    public String[] solution(String[] strings, int n) {
        String[] answer = new String[strings.length];
        List<String> list = new ArrayList<>();
        
        // 리스트 맨 앞에 n번째 문자열 붙이기
        for (int i=0; i<strings.length; i++) {
            list.add(strings[i].charAt(n) + strings[i]);
        }
        
        // 리스트 정렬
        Collections.sort(list);
        
        // 첫번째 문자 자르고 반환
        for (int i=0; i<list.size(); i++) {
            answer[i] = list.get(i).substring(1, list.get(i).length());
        }
        return answer;
    }
}