https://programmers.co.kr/learn/courses/30/lessons/12915
코딩테스트 연습 - 문자열 내 마음대로 정렬하기
문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 [sun, bed, car]이고 n이 1이면 각 단어의 인덱스 1�
programmers.co.kr
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
import java.util.Arrays;
class Solution {
private class StringSort implements Comparable<StringSort> {
private final String word;
private final int index;
public StringSort(String word, int index) {
this.word = word;
this.index = index;
}
@Override
public int compareTo(StringSort o) {
if (this.word.charAt(index) == o.word.charAt(index)) {
return this.word.compareTo(o.word);
}
return (this.word.charAt(index) - o.word.charAt(index));
}
public String getWord() {
return word;
}
}
public String[] solution(String[] strings, int n) {
int stringLength = strings.length;
StringSort[] sortedStrings = new StringSort[stringLength];
String[] answer = new String[stringLength];
for (int i = 0; i < stringLength; i++) {
sortedStrings[i] = new StringSort(strings[i], n);
}
Arrays.sort(sortedStrings);
for (int i = 0; i < stringLength; i++) {
answer[i] = sortedStrings[i].getWord();
}
return answer;
}
}
|
cs |
정렬클래스에서 정렬을해서 가져온다.
StirngSort class를 선언하고, compareTo를 override하여 구현을 한다.
sortString 배열에 값을 넣는다. compare에서 word의 index로 정렬.