Computer Engineering/알고리즘 테스트

프로그래머스 코딩테스트 문제: 문자열 내 마음대로 정렬하기

말하는호구마 2020. 3. 10. 21:31

프로그래머스 코딩테스트 level1문제이다. 

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

 

Arrays.sort 와 comparator을 자유롭게 사용한다면 쉬운 문제이다!

import java.util.*;
import java.lang.*;
import java.math.*;
class Solution {
  public String[] solution(String[] strings, int n) {
      String[] answer = strings;
      Arrays.sort(answer, new Comparator<String>(){
          public int compare(String a, String b){
              if((a.substring(n,n+1)).equals(b.substring(n,n+1))){
                  return a.compareTo(b);
              }
              return (a.substring(n,n+1)).compareTo(b.substring(n,n+1));
          }
      });
      return answer;
  }
}

 

substring은 리턴값이 String이고 charAt()은 리턴값이 char임을 기억하자

substring(a,b)는 a이상 b미만을 가져온다.

이를 이용해서 charAt()처럼 한글자만 가져올 수 있다!.