[코딩테스트 연습]level 1. 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 함수를 작성해주세요.

 

 

-------------

 

commands를 돌면서 i,j에 맞춰 array를 자르고, 정렬하여 k번째 것을 가져오는 것으로 생각을 하며 코딩을 하였다.

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

    for(let c=0; c<commands.length; c++ ){
        let i = commands[c][0]-1;
        let j = commands[c][1];
        let k = commands[c][2]-1; 

        let newArr = array.slice(i,j);
    	for(let a = 0; a<newArr.length; a++){       
        	for(let b = a+1; b<newArr.length; b++){
            	if(newArr[a] > newArr[b]){
                	let temp = newArr[b];
                	newArr[b] = newArr[a];
                	newArr[a] = temp;
            	}          
        	}
     	}  
        answer.push(newArr[k]);
    }
    return answer;
}

 

 

 

 

다른사람들의 풀이를 보았는데, 맥락은 같았지만 map을 통해 조회하고, sort를 이용한 정렬과 구조분해 문법을 사용한 것이 눈에 띄었다.

 

function solution(array, commands) {
    return commands.map(command => {
        const [sPosition, ePosition, position] = command
        const newArray = array
            .filter((value, fIndex) => fIndex >= sPosition - 1 && fIndex <= ePosition - 1)
            .sort((a,b) => a - b)    

        return newArray[position - 1]
    })
}

array의 filter로 position을 나눴는데, 이 방법보다는 내가 선택한 slice 방법을 사용했으면 더 낫지 않을까 싶다.

 

function solution(array, commands) {
    return commands.map(command => {
        const [sPosition, ePosition, position] = command
        const newArray = array
            .slice(sPosition-1, ePosition)
            .sort((a,b) => a - b)    

        return newArray[position - 1]
    })
}