본문 바로가기

프로그래머스

[프로그래머스] 문자열 내 마음대로 정렬하기_ 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 o1.charAt(n) - o2.charAt(n);
            }
        });
        
        return strings;
    }
}

 

람다식 구현 후

 

import java.util.Arrays;
import java.util.Comparator;

class Solution {
    public String[] solution(String[] strings, int n) {
        Arrays.sort(strings, (o1, o2) -> {
            if (o1.charAt(n) == o2.charAt(n)) {
                return o1.compareTo(o2);
            }
            return o1.charAt(n) - o2.charAt(n);
        });

        return strings;
    }
}