문자열 내 마음대로 정렬하기(level1)

링크

https://school.programmers.co.kr/learn/courses/30/lessons/12915

문제

문자열로 구성된 배열이 주어질 때, 오름차순 정렬을 실시하는데 주어진 숫자 n 인덱스에 위치하는 글자 순서대로 정렬을 실시하라. 인덱스 n의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치한다.

예제로 이해

주어진 배열이 ["abce", "abcd", "cdx"] 이고 주어진 숫자가 2일 경우에 2 인덱스에 위치한 문자는 순서대로 "c", "c", "x"이다 "cdx"는 가장 마지막에 위치할 것이며 문제는 "abce"와 "abcd"인데 주어진 숫자 인덱스에 위치한 문자가 같으므로 해당 문자열은 전체를 가지고 오름차순 정렬을 실시한다.

그 결과 반환 되는 배열은 ["abcd", "abce", "cdx"] 이다.

코드

function solution(strings, n) {
    strings.sort(function(a, b) {
        let A = a[n];
        let B = b[n];
        
        if(A === B) {
            if(a < b) return -1
            else return 1
        }

        if (A < B) {
          return -1;
        }
        if (A > B) {
          return 1;
        }
     });
     return strings
}

코드 풀이

기본적으로 정렬을 실시해야하므로 주어진 문자열 strings를 가지고 sort 연산을 실시한다.

정렬을 할 때 n 인덱스의 값을 기준으로 정렬을 실시해야하는데 먼저 예외에 대한 처리를 실시한다.

if(A === B) {
	if(a < b) return -1
	else return 1
}

예외는 n 인덱스의 문자가 같을 경우에 오름차순으로 정렬을 진행한다.

if (A < B) {
	return -1;
}
if (A > B) {
	return 1;
}

그 이후에 n 인덱스의 두 문자를 비교해서 정렬을 실시한다.