문제 설명
문자열로 구성된 리스트 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"] 입니다.
기초적인 알고리즘적 사고를 다시 되살리고 싶은 마음에 오래 전에 풀어봤던 문제들을 다시 풀기 시작했다.
그 시작이 "문자열 내 마음대로 정렬하기"
그다지 어려운 문제는 아니었다. strings 배열에 담긴 각 문자열 안에서 인덱스 값으로 n을 가진 문자열을 찾아 사전순으로 배열만 하면 되는 문제
다양한 풀이 방식이 존재하겠지만 이 풀이에서는 Array.prototype.sort 메서드를 사용했다.
function solution(strings, n) {
strings.sort((a, b) => {
if(a[n] === b[n]) {
return b < a ? 1 : -1
}
return b[n] < a[n] ? 1 : -1
})
return strings
}sort 메서드는 배열에 사용 가능하며 인자로 넘겨진 콜백 함수가 해당 배열의 요소들을 비교해 리턴하는 값을 토대로 정렬을 진행한다.
숫자 요소를 비교하는 경우 음수, 양수 중 어떤 값을 리턴하는가에 따라서 정렬 순서가 결정된다.
문자의 경우도 마찬가지로 양수 혹은 음수를 리턴해 주어야 하기 때문에 크기 비교를 통해 알파벳 순을 기준으로 더 앞의 글자를 판단, 삼항 연산자로 숫자를 임의로 리턴시켜 정렬한다.
위 코드를 더 축소 시키는 것도 가능할 것 같다.
function solution(strings, n) {
strings.sort((a, b) => {
return a[n] === b[n]
? a > b ? 1 : -1
: a[n] > b[n] ? 1 : -1
})
return strings
}똑같은 풀이 방식에 삼항 연산자를 조금 더 활용해 코드를 정돈해 보았다.
