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

[필요개념]

> 딱히없음. 깔끔하게 잘 푼 것 같음!

 

import java.util.ArrayList;
import java.util.Collections;

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

        //ArrayList 자료구조 list 생성
        ArrayList<String> list = new ArrayList<>();
        // list안에 입력배열값 추출한 후 안에 담아놓음
        // (단어단위 분리한 후 n번째 문자 + 문자 그 자체
        for(String string:strings){
            list.add(string.split("")[n]+string);
        }

        // 오름차순으로 정렬
        Collections.sort(list);
        // 배열크기 할당
        String[] answer = new String[list.size()];

        // list안의 값을 가져오며, substring으로 앞글자 빼고 기입함
        for(int i=0; i< answer.length; i++){
            answer[i] = list.get(i).substring(1);
        }

        return answer;
    }
}

    public class Hello {
        public static void main(String[] args) {
            Solution test = new Solution();
            String [] p = new String[]{"sun","bed","car"};
            int o = 1;
            System.out.println(test.solution(p,o));
        }
    }