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

슬슬 HTML이랑 CSS에 대한 기억이 흐릿해져 가고 있다...복습해야됨!

문제 설명

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

제한 조건

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

입출력 예stringsnreturn

["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"] 입니다.

여러 문자를 비교하여 n번째 인덱스로 정렬해주는 문제였다. 맨 처음에는 sort() 안의 함수를 어떻게 건드려서 풀어보려 했는데 영 잘 되지 않아 정공법을 택하기로 했다. 

1. strings의 요소들을 n 직전까지 slice()한 뒤 맨 뒤에 붙여줌
2. n번째 인덱스부터 끝까지 slice()하여 새로운 배열에 push()
3. 새로운 배열의 요소들은 n번째 인덱스가 맨 앞글자이므로, sort()
4. 맨 뒤에 붙인 n 전까지 slice()한 문자열들을 다시 맨 앞으로 가져옴
5. 원래 문자열의 길이만큼 slice()한 뒤 answer 배열에 push()

이러한 방법을 사용하였다. 

 

구현은 간단했다. for문을 사용하여 strings의 각 요소들에 접근하였고, 범위를 지정하여 slice()와 push()를 사용해 문자열 붙이기 밑 자르기, 그리고 sort()로 정렬 후 다시 반대로 실행하기만 하면 됐다.

var answer = [];
let arr = []
strings.sort()
for(let i = 0; i < strings.length; i++){
   strings[i] = strings[i] + (strings[i].slice(0, n))
   arr.push(strings[i].slice(n))
}
arr.sort()
for(let i = 0; i < arr.length; i++){
   arr[i] = arr[i].slice(arr[i].length - n, arr[i].legnth) + arr[i]
   answer.push(arr[i].slice(0, arr[i].length - n))
}

하지만 이 방법은 아주아주아주 커다란 문제점이 있었다. 바로 조건중 

• 인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

을 만족하지 못하는 경우 대책이 없다는 것!

예를 들어 strings가 ["cast", "bat", "dad"]인 경우,  arr 배열엔 ["astc", "atb", "add"]가 저장된다. 

하지만 이 arr 배열을 sort할 경우 a가 모두 같으므로 a 뒤의 문자열들을 기준으로 정렬하게 되는데, 각 요소에서 a 뒤의 문자는 "s", "t", "d"이므로 사전순 정렬을 할 경우 ["add", "astc", "atb"]가 된다.

이대로 2번째 for문을 실행하면 answer 배열에는 ["dad", "cast", "bat"]이 저장되는 것이다.

d > c > b이므로 위 조건에 완벽하게 부합하지 못하게 된다. 

같은 방법으로 a를 맨 뒤로 보내고 다시 정렬한 뒤 다시 a부터 붙이는 과정을 거칠 수 있겠으나, for문을 다시 사용해야 하는 불편함이 있고, 무엇보다 a 뒤의 문자도 같은 경우에는 정말 답이 없어지므로 바로 생각을 그만두었다.

 

결국 sort() 함수를 다시 만지작 거릴 수 밖에 없었다. 하지만 아무리 생각해도 답이 나오지 않아 결국 SOS....

이거 1단계 맞냐니깐...?

 

주요 내용은 sort() 함수 내 compareFunction을 잘 이용하라는 것. 그게 뭔데 이 씹덕아!

compareFunction은 sort()의 매개변수로 들어가는 함수로, 두개의 인자를 받아 그 인자를 비교하여 정렬해주는 함수이다.

let a = [3,1,2]
a.sort((앞의 수, 뒤의 수)=>앞의 수 - 뒤의 수)
console.log(a) // result = [1,2,3]

compareFunction은 두개의 인자를 가진다. 편의상 앞의 수와 뒤의 수라고 한다.

두 수는 a에 속한 요소들이다. 쉽게 말해 자동으로 for문을 돌리는 것이라고 생각하면 편할듯

그래서 앞의 수와 뒤의 수를 비교하여 결과에 따라 오름차순, 내림차순 정렬을 실시하는 것.

자세한 내용은 이쪽 참고

sort()

여튼, 이 compareFunction은 인자로 받는 두 수를 이용한 함수인데, 이게 두 인자간의 비교 뿐만 아니라 인자에 함수나 계산을 적용한 것도 비교가 가능하다. 이를 이용하여 문제를 해결하면 된다.

 

compareFunction의 두 인자를 a, b로 주고, 화살표 함수를 이용해 함수를 나타내준다.

• 만약 n번째 인덱스가 다르다면 -> n번째 인덱스 순으로 정렬
• 만약 n번째 인덱스가 같다면 -> 사전순 정렬

다음과 같은 메커니즘을 따라야 한다.

먼저 if문을 통해 n번째 인덱스가 같은지 다른지 파악해야 한다.

이 문제를 푸신 분은 아스키 코드를 통해 n번쨰 문자열이 같은지를 파악하셨다. 천재신가?

 

아스키코드를 적용하는 법은 "문자열".charCodeAt(n)이었다. 

a.charCodeAt(n)은 비교하는 문자열의 n번쨰 인덱스, b.charCodeAt(n)은 기준 문자열의 n번째 인덱스이므로, 둘이 같다면 사전순 정렬. 즉, a와 b를 직접 비교하면 된다.

a와 b의 비교법은 위의 링크에 나와있듯 a > b이면 1을 리턴, a < b이면 -1을 리턴하는 것이었다.

 

둘이 다르다면 n번째 인덱스 순으로 정렬. 단순히 a 대신 a.charCodeAt(n), b 대신 b.charCodeAt(n)을 넣어 비교하면 되는 것이었다.

 

function solution(strings, n) {
    var answer = [];
    strings.sort((a,b) => {
        if(a.charCodeAt(n) == b.charCodeAt(n)){
            if(a > b){
                return 1
            } else{
                return -1
            }
        } else{
            return a.charCodeAt(n)-b.charCodeAt(n)
        }
    })
    answer = strings
    return answer;
}

비록, 내가 직접 풀지는 못했지만 이 방법을 통해 sort()의 사용법을 좀더 깊게 이해할 수 있게 된 것 같다.

처음 봤을때 sort()를 써야지 생각만 하고 결국 사용하지 못했는데, 이 풀이를 통해 compareFunction과 두 인자의 무궁무진한 사용법에 한걸음 더 다가설 수 있게 되었다. 

 

좋은 풀이를 제공해 주신 익명의 작성자님께 무한한 감사 인사를 드립니다!