2021-07-14 / 문자열 내 마음대로 정렬하기

문제 설명

문자열로 구성된 리스트 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"] 입니다.

 

 

초기 코드

class Solution {
    public String[] solution(String[] strings, int n) {
        String[] answer = {};
        return answer;
    }
}

 

 

정답 코드

import java.util.Arrays;
class Solution {
    public String[] solution(String[] strings, int n) {
        int len=strings.length;
        String[] answer =new String[len];

        for(int i=0;i<len; i++){
          strings[i]=strings[i].charAt(n)+strings[i];
        }
        Arrays.sort(strings);
        for(int i=0; i<len; i++){
            answer[i]=strings[i].substring(1);
        }
        return answer;
    }
}

 

 

코드 설명

문자열 중 n번째 인덱스의 문자를 가지고 정렬을 하는 문제이다. 

먼저 n번째의 문자를 가져온다. (charAt(n))
문자를 가져와 문자열에 맨 앞에 붙인 후 (strings[i]=strings[i].charAt(n)+strings[i])정렬을(Arrays.sort(strings)) 하게 되면 n번째의 문자가 중복되더라도 사전순으로 앞선 문자열이 앞쪽에 위치하게 된다.
answer의 방에 맨 앞 문자를 뗀 후 원래 형태의 문자열을 넣는다.  (answer[i]=strings[i].substring(1))

 

 

 

---

 

 

 

 

 

 

 

사실 처음에는 n번째의 문자를 가져와 따로 char 형태의 배열에 넣고 정렬을 했었다.  그후 해당되는 문자가 있는 문자열을 answer방에 넣었는데 이럴 경우 n번째 문자가 중복되는 경우 실패했다. 그래서 다른 사람의 머리를 빌려.. 문제를 풀게 되었다.

 

 

 

 

 

코딩테스트 연습 - 문자열 내 마음대로 정렬하기