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]
}
}