코딩 테스트
(Java) 프로그래머스 문자열 내 마음대로 정렬하기
로승리
2021. 11. 5. 02:30
리스트 맨 앞에 n번째 문자열을 붙이는게 핵심 포인트였다.
처음에는 문자열을 split올 다 잘라서 정렬하려고 했는데 생각처럼 잘 정렬이 되지 않았다.
검색을 통해 힌트를 얻어 해결했다.
최종 코드
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
class Solution {
public String[] solution(String[] strings, int n) {
String[] answer = new String[strings.length];
List<String> list = new ArrayList<>();
// 리스트 맨 앞에 n번째 문자열 붙이기
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;
}
}