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

---

import java.util.*;

class Solution {
    public String[] solution(String[] strings, int n) {
        
        Arrays.sort(strings, new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                if(o1.charAt(n) == o2.charAt(n)) {
                    return o1.compareTo(o2);
                }
                return Character.compare(o1.charAt(n), o2.charAt(n));
            }
        });

        return strings;
    }
}

 

처음에는 HashMap으로 풀어보려했는데 HashMap 같은 경우는 중복 키가 없기 때문에 만약 값은 문자가 들어오면 값이 덮어쓰여질 수 있다. 

그리고 HashMap은 원래 삽입 순서를 보장하지 않아서 

comparator 를 사용해서 풀었다