[JS] 프로그래머스 : K번째 수

 

Question

프로그래머스 : K번째수

 

 

입출력 예시

 

 

입출력 예 설명

 

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

 

 

---

 

 

My Code

 

1. for문을 돌려서 answer에 push 하는 코드

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

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

 

정답이긴 한데 어쩐지 코드가 지저분해 보인다.

 

HOW? (풀이방법)

 

• 정답인 배열은 총 commands의 길이의 수와 똑같으므로 for문을 그만큼 돌려준다.
• commands의 조건에 맞는 배열을 slice를 통해 형성한 후 sort로 오름차순 정렬을 해준다.
• 정렬된 배열의 n번째 수를 arr에 push 한다.

 

*** 컴퓨터는 배열을 0부터 센다. 따라서 범위의 수를 -1 한다.

*** slice(n, m)은 인덱스 n부터 m-1까지를 새로운 배열로 반환해주므로 slice앞부분 범위만 -1을 한다.

 

 

 

2. commands의 map을 사용한 코드

function solution(array, commands) {
    let answer = commands.map((ele) =>
        array
        .slice(ele[0] - 1, ele[1])
        .sort((a, b) => a - b)[ele[2] - 1])

    return answer;
}

 

HOW? (풀이방법)

 

• for문에서 돌던 commands의 요소들을 map을 이용해서 새로운 배열을 반환했다.
• 각각 요소들의 접근으로 이차원 배열의 접근이 일차원 배열로 바꿨다.
• slice를 하고 sort를 사용하는 방식은 똑같다.

 

더 깔끔한 코드가 됐다!!!!

문제를 푸는 접근은 비슷하지만 어떤 방식을 택하느냐에 따라 가독성이 달라졌다.

 

 

 

 

풀이 시간의 차이는

첫 번째 푸는 방식

 

 

두 번째 푸는 방식

 

케이스 바이 케이스에 따라 더 빠르거나 느린 거 같아 보인다.

근데 별로 차이 안나 보인다.