Lv.1 문자열 내 마음대로 정렬하기

https://programmers.co.kr/learn/courses/30/lessons/12915

코딩테스트 연습 - 문자열 내 마음대로 정렬하기

기본틀

class Solution {
    public String[] solution(String[] strings, int n) {
        String[] answer = {};
        return answer;
    }
}

첫번째 시도

import java.util.*;
class Solution {
    public String[] solution(String[] strings, int n) {
        
        Arrays.sort(strings);
        
        String[] arr1 = new String[strings.length];
        for(int i = 0; i < arr1.length; i++){
            arr1[i] = strings[i].substring(n, n+1);
        }
        
        Set hs = new HashSet();
        for(int i = 0; i < arr1.length; i++){
            hs.add(arr1[i]);
        }
        
        String[] arr2 = new String[hs.size()];
        
        Iterator<String> it = hs.iterator();
        int i = 0; 
        while(it.hasNext()){
            arr2[i] = it.next();
            i++;
        }
        
        Arrays.sort(arr2);
        
        String[] answer = new String[arr1.length];
        
        
        int k = 0;
        for(int h = 0; h < arr2.length; h++){
            for(int j = 0; j < arr1.length; j++){
                if(arr2[h].equals(strings[j].substring(n, n+1))){
                    answer[k] = strings[j];
                    k++;
                }
            }
        }
        
        
        return answer;
    }
}

성공~ 힘들었당;;