프로그래머스 정렬 : k번째 수 - 자바스크립트

 

문제 설명
배열 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		
[1, 5, 2, 6, 3, 7, 4]	

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

return
[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입니다.

 

✍️  풀이과정 

// 프로그래머스 정렬 k번째 수 

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

let newArray = [];
let result = [];

function solution(array, commands) {

 for(i = 0; i < commands.length; i++){
    console.log("i단 : " + i);
    newArray = commands[i];
    console.log("newArray : " + newArray);

		// 1단계 자르기 
    let sliceArray = array.slice(newArray[0] - 1,newArray[1]);
    console.log("sliceArray : " + sliceArray)
		
		// 2단계 정렬
    let sortArray = sliceArray.sort();
    console.log("sortArray : " + sortArray);
		
		// 3단계 결과값
    let resultArray = sortArray[newArray[2]-1]
    console.log("resultArray : " + resultArray)
		
		// 4단계 결과값 배열에 저장 
    result.push(resultArray);
 }
	
	// 5단계 리턴 
  return console.log("최종 result : " + result);

}

 

최종 코드

// 테스트 코드 
let array = [1, 5, 2, 6, 3, 7, 4];
let commands = [[2, 5, 3], [4, 4, 1], [1, 7, 3]];

function solution(array, commands) {

let newArray = [];
let result = [];

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

 

 

이유는 ? 

-> 펑션 추가하지 않으면 문자열로 인식하여 정렬함.

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

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

 

- 관련 레퍼런스 

Array.prototype.sort() - JavaScript | MDN

 

🏃‍♂️ 최종 통과 코드!

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

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

 for(var i = 0; i < commands.length; i++){
    newArray = commands[i];
    let sliceArray = array.slice(newArray[0] - 1,newArray[1]);

    let sortArray = sliceArray.sort((a, b) =>{
        return a - b;
    });

    let resultArray = sortArray[newArray[2]-1]
    result.push(resultArray);
 }
  return result ;
}

 

- 다른 풀이 : 리팩토링 코드 

function solution(array, commands) {
    return commands.map(v => {
        return array.slice(v[0] - 1, v[1]).sort((a, b) => a - b).slice(v[2] - 1, v[2])[0];
    });
}

 

 

---

깨달은 점 

- 자바스크립트는 자바와 다르게 sort 함수를 사용하면 문자열로 취급하여 정렬하기 때문에 내 예상과 다르게 정렬될수 있다.

- 화살표 함수를 사용해서 function을 추가해주는게 키포인트