programmers 문자열 내 마음대로 정렬하기

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

코딩테스트 연습 - 문자열 내 마음대로 정렬하기

 

 

문제 분석

- Comparator 익명 클래스를 활용하여 정렬 기준을 설정한다.

- compare메소드를 Override하여 n번째 글자를 기준으로 비교하고, 만약 n번째 글자가 동일하다면 사전순으로 정렬한다. 

 

 

문제 풀이

1. Comparator 익명 클래스를 생성한다.

2. compare 메소드를 Override한다.

3. if/else문을 통해 n번째 글자가 같은 경우 글자를 사전순으로 정렬하고, 다를 경우 n번째 인덱스를 기준으로 정렬한다.

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 s1.compareTo(s2);
                }else{
                    return s1.charAt(n) - s2.charAt(n);
                }
            }
        });
            
        String[] answer = new String[strings.length];
        answer = strings;
        return answer;
    }
    
}

 

 

https://github.com/SOEUN2/Algorithm