Front-End

vsoghlv@naver.com

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

문제 설명

문자열로 구성된 리스트 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]

문제풀이

function solution(strings, n) {
  var answer = [];
  //두번째 값 추출
  let secondIndex = [];
  for (let i = 0; i < strings.length; i++) {
    secondIndex.push(strings[i].substr(n, 1));
  }
  //console.log(secondIndex);
  //console.log(strings)
  secondIndex.sort();
  strings.sort();
  //다시 돌면서 값 같은 것 넣기
  for (let i = 0; i < secondIndex.length; i++) {
    for (let j = 0; j < strings.length; j++) {
      if (secondIndex[i] === strings[j].substr(n, 1)) {
        answer.push(strings[j]);
      }
    }
  }
  return answer.filter((e, index) => {
    return answer.indexOf(e) === index;
  });
}

먼저 빈배열을 만들어 받은 strings 각 값의 두번째 글자들을 추출해 넣어줬다. 그 후 정렬을 위해 strings 와 secondIndex 둘다 .sort 를 해주고 다시 이중 포문을 돌며 secondIndex 의 값과 strings 의 값들 중 두번째 값이 일치하는 것들을 넣어주고 리턴하기 전 중복되는 값들을 삭제해줬다.

오늘도 코드가 매우 길다. 언제쯤 짧고 간결하게 쓸 수 있을까.