[Programmers] 내 마음대로 문자열 정렬하기

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

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

 

 

Java

 

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

public 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 1;
				else if (o1.charAt(n) == o2.charAt(n))
					return o1.compareTo(o2);
				else
					return -1;
			}
		});
		return strings;
    }

}