[프로그래머스][JAVA]Level 1 : 문자열 내 마음대로 정렬하기

 

 

 

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

프로그래머스

 

 

 

 

1번 풀이

import java.util.*;
class Solution {
    public String[] solution(String[] strings, int n) {
        String[] answer = new String[strings.length];

        List<String> list = new ArrayList<>();
        for(int i=0; i<strings.length; i++){
            list.add(strings[i].charAt(n)+strings[i]);
        }
        Collections.sort(list);
        for(int i=0; i<list.size(); i++){
            answer[i] = list.get(i).substring(1, list.get(i).length());
        }
        return answer;
    }
}

 

처음엔 이렇게 풀었었는데..그러니까 어차피 비교문자가 같을 경우 전체 문자열을 기준으로 비교하니까, 

비교문자를 맨 앞에 두고 문자열을 그대로 합쳐주고 정렬(sort)를 하고

정렬된 리스트에서 맨 처음 문자만 빼고 반환해주는것이다.

 

근데 이렇게 말고 compare를 쓰는 방법을 찾아보았다.

 

 

2번 풀이

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 if(s1.charAt(n) < s2.charAt(n)) return -1;
              else return 0;
          }
      });
      return strings;
  }
}

 

중간 코드를 보면 

return 1 : 두 요소를 서로 바꿔야 한다는것을 나타낸다. 즉, 첫번째 요소가 두번쨰 요소보다 큰 경우에 사용된다. 이렇게 하면 두 요소가 서로 바뀌어서 오름차순으로 정렬된다.

return -1 : 두 요소의 순서를 변경할 필요가 없는 것이다. 즉, 첫번째 요소가 두번째 요소보다 작은 경우에 사용된다. 이 경우 두 요소의 순서가 이미 올바르게 정렬되어 있다는 의미이다.