[Programmers][Level1] 문자열 내 마음대로 정렬하기

울부짖는 늑대·2023년 7월 25일

프로그래머스 Coding-TEST

목록 보기

1/23

문제 설명

문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 ["sun", "bed", "car"]이고 n이 1이면 각 단어의 인덱스 1의 문자 "u", "e", "a"로 strings를 정렬합니다.

제한 사항

strings는 길이 1 이상, 50이하인 배열입니다.
strings의 원소는 소문자 알파벳으로 이루어져 있습니다.
strings의 원소는 길이 1 이상, 100이하인 문자열입니다.
모든 strings의 원소의 길이는 n보다 큽니다.
인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

입출력 예

numbers	n	result
["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"] 입니다

나의 코드

import java.util.*;
import java.util.stream.Collectors;
class Solution {
    public String[] solution(String[] strings, int n) {
        Arrays.sort(strings);
        return Arrays.stream(strings).sorted(Comparator.comparing(s -> s.charAt(n))).collect(Collectors.toList()).toArray(new String[strings.length]);
    }
}

풀이 방법

처음에는 배열 객체로 그대로 루프를 여러번 돌려 풀 생각이였지만 for문을 생각보다 많이 사용하게 될 것 같아서 Java 8 버전의 Stream을 사용하여 풀어냈다.
이 문제에서 주의깊게 봐야 할 지점은 입력받은 인덱스의 문자가 같다면 사전순으로 앞선 문자열이 앞쪽에 위치한다.
그러기에 미리 입력받은 문자열 배열을 오름차순으로 정렬해주고 stream으로 변환하여
입력받은 인덱스 n을 문자로 하나씩 뽑아내서 재정렬을 해줬다.

다른 풀이

import java.util.*;
class Solution {
    public String[] solution(String[] strings, int n) {
    Arrays.sort(strings);
        Arrays.sort(strings, new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                return o1.charAt(n) - o2.charAt(n);
            }
        });
        return strings;
    }
}