(Python, Java) 프로그래머스 - 문자열 내 마음대로 정렬하기

03 Jan 2022 in Algorithm on Programmers, Level1

[문제 링크]

Python 풀이

def solution(strings, n):
    strings.sort(key=lambda x:(x[n],x))
    return strings

Java 풀이

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 Character.compare(o1.charAt(n), o2.charAt(n));
            }
        });
        return strings;
    }
}