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

문제 링크

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

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

문제 풀이

간단한 문제지만 Comparator를 이용하여 정렬해 풀어보았다.

 

소스 코드

import java.util.*;

class Comp implements Comparator<String>{
	int n;
	
	public Comp(int n) {
		this.n = n;
	}

	public int compare(String o1, String o2) {
		if(o1.charAt(n) < o2.charAt(n) ) return -1;
		else if(o1.charAt(n) > o2.charAt(n) ) return 1;
		else return o1.compareTo(o2);
	}
}

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