[알고리즘] 문자열 내 마음대로 정렬하기

문제 설명

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

제한 조건

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

입출력 예시

🖊 풀이

📌 sort()

• sort의 작동원리를 간단하게 짚고 넘어가보겠다(자세한 내용은 따로 포스팅으로 다뤄볼 것)
• arr.sort([compareFunction])
• sort는 배열의 요소를 일시적으로 문자열로 변환한 후 정렬하므로 숫자 요소를 정렬할 때는 정렬 순서를 정의하는 비교 함수를 인수로 전달해야 한다.
  • compareFunction(a, b)이 < 0 : a가 b보다 앞에 있어야 한다.
  • compareFunction(a, b)이 = 0 : a와 b를 서로에 대해 변경하지 않고 모든 다른 요소에 대해 정렬한다.
  • compareFunction(a, b)이 > 0 : b가 a보다 앞에 있어야 한다.
• 내림차순으로 정렬을 할 경우에는 a - b를 b - a로 바꿔주면 된다.

📌 localeCompare()

• 기준 문자열과 비교했을 때 비교 대상 문자열이 정렬상 전에 오는지, 후에 오는지, 같은 순서에 배치되는 지를 알려주는 순서를 리턴한다.
• 코드 상으로는 인자로 받는 n번째 문자가 같으면 전체 문자열을 비교하고, n번째 문자가 같지 않으면 n번째 문자를 비교해서 정렬하는 방법으로 사용했다.
  
  

💡 코드

function solution(arr, n) {
  const result = arr.sort((a, b) => {
    if (a[n] > b[n]) {
      return 1;
    } else if (a[n] < b[n]) {
      return  -1;
    }
    if (a[n] === b[n]) {
      if (a > b) {
        return 1;
      } else if (a < b) {
        return - 1;
      }
      return 0;
    }
  });
  
  return result;
}

• localeCompare메서드를 사용한 코드

function solution(arr, n) {
  
  const result =  arr.sort((a, b) => { 
    if (a[n] === b[n]) {
      return a.localeCompare(b);
    } else {
      return a[n].localeCompare(b[n]);
    }
  });
  
  return result;
}

문제 출처: 프로그래머스
참조: MDN