https://programmers.co.kr/learn/courses/30/lessons/12915
문제 설명
문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 ["sun", "bed", "car"]이고 n이 1이면 각 단어의 인덱스 1의 문자 "u", "e", "a"로 strings를 정렬합니다.
제한 조건
strings는 길이 1 이상, 50이하인 배열입니다.
strings의 원소는 소문자 알파벳으로 이루어져 있습니다.
strings의 원소는 길이 1 이상, 100이하인 문자열입니다.
모든 strings의 원소의 길이는 n보다 큽니다.
인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.
입출력 예
| strings | n | return |
|---|---|---|
| ["sun", "bed", "car"] | 1 | ["car", "bed", "sun"] |
| ["abce", "abcd", "cdx"] | 2 | ["abcd", "abce", "cdx"] |
입출력 예 설명
입출력 예 1
"sun", "bed", "car"의 1번째 인덱스 값은 각각 "u", "e", "a" 입니다. 이를 기준으로 strings를 정렬하면 ["car", "bed", "sun"] 입니다.
입출력 예 2
"abce"와 "abcd", "cdx"의 2번째 인덱스 값은 "c", "c", "x"입니다. 따라서 정렬 후에는 "cdx"가 가장 뒤에 위치합니다. "abce"와 "abcd"는 사전순으로 정렬하면 "abcd"가 우선하므로, 답은 ["abcd", "abce", "cdx"] 입니다.
풀이
- Comparator 구현(속도 훨씬 빠름)
* Comparator와 Comparable
여기서는 두 수를 비교해야 하기 때문에 comparator를 사용.
- compareTo() : 두 개의 값 비교하여 int 값으로 반환해주는 함수.
- 문자 기준 아스키 값으로 비교한다.
- 대소문자를 구분한다.
- 대소문자 무시하고 비교하고 싶다면 compareToIgnorecase()를 사용하면 된다.
import java.util.*;
class Solution {
public String[] solution(String[] strings, int n) {
Arrays.sort(strings, new Comparator<String>() {
@Override
public int compare(String o1, String o2) {
char ch1 = o1.charAt(n);
char ch2 = o2.charAt(n);
// compareTo는 두 개의 값을 비교하여 int값으로 반환해주는 함수. (숫자와 문자비교)
// 문자기준 아스키값을 기준으로 비교한다.(대소문자구분.);
if(ch1 == ch2) {
return o1.compareTo(o2);
} else return ch1 - ch2;
}
});
return strings;
}
}- ArrayList이용
word의 n번째 문자열 빼서 word앞에 붙인 후 오름차순으로 정렬.
import java.util.*;
class Solution {
public String[] solution(String[] strings, int n) {
//string배열은 수정 불가하여 ArrayList로 변환.
ArrayList<String> list = new ArrayList<>();
for(String word:strings) {
list.add(word.charAt(n)+word);
}
// 정렬방법 2가지 list.sort(Comparator.naturalOrder());
Collections.sort(list);
String[] answer = new String[strings.length];
// 추가한 문자열 제거
for(int i = 0; i < strings.length; i++) {
answer[i] = list.get(i).substring(1);
}
return answer;
}
}학습
🧨 정렬
정렬할 때는 항상 기준이 필요하다.
sort() 사용할 때 아무런 정렬기준을 주지 않았는데도 사용가능한 이유는 java내부에 정렬 기준을 가지고 있기 때문이다.
컬렉션을 정렬하는데 필요한 메서드를 정의하는 Comparator와 Comparable 인터페이스를 가지고 있다.
❓Comparator
기본 정렬기준을 구현하는데 사용.
❓Comparable : 기본 정렬기준 외에 다른 기준으로 정렬하고자 할 때 사용.
