[프로그래머스] 문자열 내 마음대로 정렬하기 - JAVA

[작성일: 2024. 03. 09]

 

 

풀이

import java.util.Arrays;

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

 

strings 문자열 배열에서 문제에 주어진 n 인덱스 + strings[i]를 한 문자열을 list[i]에 저장한다.

예를 들어 strings[i]에 abcd가 저장되어 있고, n이 3이라면 list[i]에는 dabcd가 저장된다.

Arrays.sort를 사용하면 자연스럽게 n번째 인덱스를 기준으로 문자열을 정렬할 수 있게 된다.

answer에 문자열을 저장할 때는 첫번째 문자를 제외하고 저장하기 위해 substring(1)을 사용했다.