[프로그래머스] 연습문제 (문자열 내 마음대로 정렬하기)

문제 설명

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

 

나만의 해설

들어온 값의 n번째 문자를 기준으로 정렬을 하라 했으니 charAt으로 뽑아서

기존의 들어온 값 맨 앞에 붙여버리고 그 상태에서 정렬해주었다
다시 반복문 돌며 붙인거 이후 인덱스인 1 부터 문자열 끝 인덱스 까지 잘라

answer에 대입해 주었다

 

public class Solution {
    public static void main(String[] args) {
        String[] answer = {};
        String[] strings = {"sun", "bed", "car"};
        int n = 1;

        ArrayList<String> arr = new ArrayList<>();
        for (int i = 0; i < strings.length; i++) {
            arr.add("" + strings[i].charAt(n) + strings[i]);
        }
        Collections.sort(arr);
        answer = new String[arr.size()];
        for (int i = 0; i < arr.size(); i++) {
            answer[i] = arr.get(i).substring(1, arr.get(i).length());
        }
        System.out.println(answer);
    }
}