[프로그래머스 Lv.1] k번째 수

Soozynn·2021년 9월 17일
0

알고리즘 풀이

목록 보기
5/6

문제 설명

배열 array의 i번째 숫자부터 j번째 숫자까지 자르고 정렬했을 때, k번째에 있는 수를 구하려 합니다.

예를 들어 array[1, 5, 2, 6, 3, 7, 4], i = 2, j = 5, k = 3이라면

array의 2번째부터 5번째까지 자르면 [5, 2, 6, 3]입니다.
1에서 나온 배열을 정렬하면 [2, 3, 5, 6]입니다.
2에서 나온 배열의 3번째 숫자는 5입니다.

배열 array, [i, j, k]를 원소로 가진 2차원 배열 commands가 매개변수로 주어질 때, commands의 모든 원소에 대해 앞서 설명한 연산을 적용했을 때 나온 결과를 배열에 담아 return 하도록 solution 함수를 작성해주세요.

<제한사항>

  • array의 길이는 1 이상 100 이하입니다.
  • array의 각 원소는 1 이상 100 이하입니다.
  • commands의 길이는 1 이상 50 이하입니다.
  • commands의 각 원소는 길이가 3입니다.

<입출력 예>
array / commands / return
[1, 5, 2, 6, 3, 7, 4] / [[2, 5, 3], [4, 4, 1], [1, 7, 3]] / [5, 6, 3]

<입출력 예 설명>

  • [1, 5, 2, 6, 3, 7, 4]를 2번째부터 5번째까지 자른 후 정렬합니다. [2, 3, 5, 6]의 세 번째 숫자는 5입니다.
  • [1, 5, 2, 6, 3, 7, 4]를 4번째부터 4번째까지 자른 후 정렬합니다. [6]의 첫 번째 숫자는 6입니다.
  • [1, 5, 2, 6, 3, 7, 4]를 1번째부터 7번째까지 자릅니다. [1, 2, 3, 4, 5, 6, 7]의 세 번째 숫자는 3입니다.



<처음 작성했던 코드>

function solution(array, commands) {
    const result = [];

    for (let i = 0; i < commands.length; i++) {
        let command = commands[i];
        let sliceArray = array.slice(command[0] - 1, command[1]);
        let sortArray = sliceArray.sort();
        
        result.push(sortArray[command[2] - 1]);
    }
    return result;
}

아래와 같이 샘플 테스트는 통과하였으나..

채점을 돌려보니 테스트 2에서 시간초과로 걸리는 듯 하다..

그래서 코드를 다시 짜보았다.

function solution(array, commands) {
    const result = [];
    
       for (let i = 0; i < commands.length; i++) {
       	  const command = commands[i];
          const sliceArray = array.slice(command[0] - 1, command[1]);
          
          const sortArray = sliceArray.sort(function (a,b) {
          	return a - b;
      });
      
        result.push(sortArray[command[2] - 1]);
    }
    return result;
}

이번에는 모든 테스트를 잘 통과하였다. 무엇이 문제였을까?🤔

첫 번째 코드 작성 했을 때 문제점 ↓)

sort() -> 해당 메서드 작성 시 아래 문법과 같이 함수를 정의 안해주었으므로 내가 생각한 것처럼 오름차순으로 값이 정렬되지 않았던 것.
아래 MDN예시를 보면 더 자세히 알 수 있다.

그렇다면 어떻게 작성하면 좋을까?


sort()

배열의 요소를 적절한 위치에 정렬한 후 그 배열을 반환한다.

Syntax

arr.sort([compareFunction])

기본 구조를 보면, compareFunction으로 정의된 정렬 순서를 통해 arr를 정렬하여 반환하는 것인데, 처음 코드를 짰을 때 나는 compareFunction의 값을 주지 않았으므로 코드가 제대로 구현되지 않았던 것 같다.
compareFunction을 생략하면 기본적으로 각 요소의 유니코드 값에 따라 자동으로 정렬된다.

정렬되면서 기존의 배열 데이터가 변경되는 메서드이기 때문에 신중하게 사용해야한다.

const months = ['March', 'Jan', 'Feb', 'Dec'];
months.sort();
console.log(months);
// expected output: Array ["Dec", "Feb", "Jan", "March"]

const array1 = [1, 30, 4, 21, 100000];
array1.sort();
console.log(array1);
// expected output: Array [1, 100000, 21, 30, 4]

위에 코드를 보면 문자 정렬은 알파벳 순서대로 잘 정렬이 된 것을 확인할 수 있지만,

숫자는 유니코드 순서로 정렬하기 때문에 결과 값이 우리가 생각했던 순서로 정렬되지 않음을 확인했다.

다음은 위의 오류를 개선하기 위한 코드이다.

1) 오름차순

numbers.sort(function (a, b) {
	return a - b;
});
console.log(numbers);
// result -> [ 1, 4, 21, 30, 10000 ]

2) 내림차순

numbers.sort(function (a, b) {
	return b - a;
});
console.log(numbers);
// result -> [ 10000, 30, 21, 4, 1 ]

해당 코드들은 화살표 함수를 사용하여 짧고 간결하게 변환 가능하다.

numbers.sort((a, b) => a - b); // 오름차순
numbers.sort((a, b) => b - a); // 내림차순

하지만, sort() 로 작성해주어도 테스트에서는 1의 자리만 존재하였기 때문에 자동으로 오름차순으로 정리될 것이라 생각하였는데 결과는 그렇지 않았다. 이에 대해서는 더 조사해보아야 할 것 같다.

출처)

0개의 댓글