[LeetCode] Kth Smallest Element In A Sorted Matrix

문제링크

Kth Smallest Element in a Sorted Matrix - LeetCode

정렬된 2차 행렬을 1차 행렬로 변환했을때 k번째 수를 구하는 문제

접근 방법

1. 2차 행렬을 1차 행렬로 변환하기

2. 변환된 1차 배열을 정렬하기

3. k번째 수를 구하기

라는 간단한 로직으로 풀 수 있다.

// 2차 배열을 1차 배열로 변환하는 방법 1 (spread)

const matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]];

let mergeArr = [];

for(let arr of matrix){
	mergeArr = [...mergeArr, ...arr]
}​

 

// 1차 배열을 2차 배열로 변환하는 방법 2 (concat)

const matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]];

let mergeArr = [];

for(let arr of matrix){
	mergeArr = mergeArr.concat(arr);
}

 

// 1차 배열을 2차 배열로 변환하는 방법 3 (flat)

const matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]];

const mergeArr = matrix.flat()​

// flat 메서드 사용

/**
 * @param {number[][]} matrix
 * @param {number} k
 * @return {number}
 */
var kthSmallest = function(matrix, k) {
    let mergeArr = [];
    mergeArr = matrix.flat();
    
    mergeArr.sort((a, b) => a - b);
    
    return mergeArr[k - 1]
};