프로그래머스 알고리즘 k번째 수

문제 설명

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

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

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

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

제한사항

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

입출력 예

array = [1, 5, 2, 6, 3, 7, 4]

commands = [[2, 5, 3], [4, 4, 1], [1, 7, 3]]

return = [5,6,3]

sort((a,b) => {return a-b}을 하는 이유

[JavaScript] .sort(function(a, b) { return a - b; })가 작동하는 원리?

• sort함수는 기본적으로 배열을 문자열 타입으로 간주해주기 때문인데, [90,100,110]을 sort()로 정렬하면 자바스크립트 값을 “90” “100” “110”으로 간주하고 비교한다. 이때 “90”과 “100”을 비교 하게 되면 앞자리 9와 1을 비교하게 되서, 100보다 90이 더 크다는 결론을 내리게된다.
• 따라서 별도록 정의된 compare function을 이용해야 한다.
• function(a,b){return a-b}라는 함수는 a와 b의 값을 빼고 값이 양수가 나오면 배열에서 자리를 교체하고 음수가 나오면 교체하지 않는다

문제 풀이 해결 아이디어

• for문을 commands.length 만큼 돌면서 commands의 0번째와 1번째수로 slicing을 해준다.
• slicing을 해준 배열을 오름차순으로 정렬해준다. 이때 정렬은 (a,b) ⇒ {return a-b} 로 진행해준다.
• 이렇게 slicing을 해준 변수에서 k번째 값을 뽑아내준다. k번째 값은 commands[i][2]에 존재한다.
• array.push기능을 이용하여 k번째 값을 차례대로 넣어준다.

소스 코드

function solution(array, commands) {
    var answer = []; 
    
    for(var i = 0;i < commands.length ;i++){
        var element = array.slice(commands[i][0]-1,commands[i][1]).sort((a,b) => {return a-b})
        answer.push(element[commands[i][2]-1])
    }
    return answer;
}