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

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

프로그래머스

---

인덱스를 똑 가져와서 비교 후 정렬했다.

 

- Java

import java.util.*;
class Solution {
    public String[] solution(String[] ss, int n) {
        Arrays.sort(ss, new Comparator<String>() {
        @Override
        public int compare(String o1, String o2) {
            String o11 = String.valueOf(o1.charAt(n));
            String o22 = String.valueOf(o2.charAt(n));
            if(o11.compareTo(o22)==0)
                return o1.compareTo(o2);
            else return (o11.compareTo(o22));
        }
        });
        return ss;
    }
}

 

- Ruby

루비는 sort_by라는 함수로 간단하게 구현할 수 있다.

신기방기...

def solution(s, n)
    s.sort!
    s.sort_by!{|a| a[n]}
    return s
end