[프로그래머스] 코딩테스트 연습 - 연습문제 Level 1 문자열 내 마음대로 정렬하기

uoahy·2021년 9월 27일
0

Solution.java

import java.util.*;

class Solution {
    public String[] solution(String[] strings, int n) {
        String[] answer = {};
        
        Arrays.sort(strings, new MyComparator(n));
        
        answer = strings;
        
        return answer;
    }
}

class MyComparator implements Comparator<String> {
    int n;
    
    MyComparator(int n) { this.n = n; }
    
    public int compare(String s1, String s2) {
        if (s1.charAt(n) != s2.charAt(n)) return s1.charAt(n) - s2.charAt(n);
        return s1.compareTo(s2);
    }
}

출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges

0개의 댓글