프로그래머스 LV1 - 문자열 내 마음대로 정렬하기

문자열 내마음대로 정렬하기 Go

문제설명

문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 ["sun", "bed", "car"]이고 n이 1이면 각 단어의 인덱스 1의 문자 "u", "e", "a"로 strings를 정렬합니다.

제한사항

strings는 길이 1 이상, 50이하인 배열입니다.
strings의 원소는 소문자 알파벳으로 이루어져 있습니다.
strings의 원소는 길이 1 이상, 100이하인 문자열입니다.
모든 strings의 원소의 길이는 n보다 큽니다.
인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

생각을 해보자

Collections.sort의 Comparator 인터페이스 구현해서 할수있겠다..
Comparator 문법 다까먹었으니 구글링 -> Comparable 참고자료
Comparator : compare()의 리턴 값이 음수가 리턴되면 오른쪽 인자가 뒤로

내풀이

import java.util.*;
class Solution {
    public String[] solution(String[] strings, int n) {
        String[] answer = {};
        
        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;
    }
}

• s1.compareTo(s2) :
  c++에서는 바로 s1 > s2가 가능하지만 자바는 불가능 ^_ㅠ
  --> 자바에서 문자열 비교시 compareTo() 함수사용
  s1이 s2보다 사전순으로 앞서면(값이 작으면) 음수리턴
  s1이 s2보다 사전순으로 뒤면(값이 크면) 양수리턴
  s1 == s2 0리턴

다른사람 풀이

 Arrays.sort(strings, new Comparator<String>() {
          @Override
          public int compare(String s1, String s2){
              char c1 = s1.charAt(n);
              char c2 = s2.charAt(n);
              
              if(c1 == c2){
                  return s1.compareTo(s2);
              } else return c1 - c2;
          }
      });
      

c1 > c2 면 c1이 오른쪽으로 내려와야하고
c1 < c2 면 c2가 오른쪽으로 내려와야하니
그냥 return c1-c2 하면 되네..!!!