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

 

 

💡 나의 풀이

어려웠다..n번째 단어를 각 단어 앞에 붙여주고 오름차순 정렬해준 뒤 붙여놨던 n번째 단어를 substring으로 잘라내어 주었다. 검색해보니까 Comparator를 사용하던데 그러면 메소드 추가에 할게 많길래 나중에 더 공부해보는걸로...

 

import java.util.*;
 
class Solution {
  public String[] solution(String[] strings, int n) {
      String[] answer = {};
      
     List<String> arr = new ArrayList<String>();
      
      //charAt(n)으로 n번째 단어 + i번째 단어를 붙여준다.
      for(int i = 0; i < strings.length; i++){
          arr.add(strings[i].charAt(n) + strings[i]);
      }
      
      Collections.sort(arr);
      answer = new String[arr.size()];
      
      for(int i = 0; i < answer.length; i++){
          answer[i] = arr.get(i).substring(1, arr.get(i).length());
      }      
            
      return answer;
  }
}