[프로그래머스 Level 1] 문자열 내 마음대로 정렬하기(Java)

문제

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

프로그래머스

 

코드

import java.util.*;
class Solution {
    public String[] solution(String[] strings, int n) {
        String[] answer = {};
        Arrays.sort(strings, new Comparator<String>(){
            @Override
            public int compare(String s1, String s2){
                if(s1.charAt(n)>s2.charAt(n)){ //A>B -> B A 정렬 (A가 뒤로가게)[오름 차순]
                    return 1;
                }else if(s1.charAt(n)<s2.charAt(n)){ // A<B -> B A 정렬 (A가 앞으로가게)[오름차순]
                    return -1;
                }else{ //인덱스 n의 문자가 같다면, 전체문자열 기준으로 사전순으로 정렬
                    return s1.compareTo(s2);
                }
            }
        });
        return strings;
    }
}