Java

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

Daeryuk Kim 2024. 3. 2. 18:50
import java.util.*;

class Solution {
    public String[] solution(String[] strings, int n) {
        Arrays.sort(strings, new Comparator<String>() {
            @Override
            public int compare(String s1, String s2) {
                if (s1.charAt(n) > s2.charAt(n)) {
                    return 1;
                } else if (s1.charAt(n) == s2.charAt(n)) {
                    return s1.compareTo(s2);
                } else {
                    return -1;
                }
            }
        });
        return strings;
    }

    public static void main(String[] args) {
        Solution sol = new Solution();
        
        String[] strings1 = {"sun", "bed", "car"};
        int n1 = 1;
        System.out.println(Arrays.toString(sol.solution(strings1, n1))); 
        // 출력: [car, bed, sun]

 
    }
}