본문 바로가기
Algorithm

[Algorithm][Recursion] 문제 풀이 #22 - Kth Smallest Element in a Sorted Matrix

by Lee David 2022. 8. 13.
반응형
문제 링크

https://leetcode.com/problems/kth-smallest-element-in-a-sorted-matrix/

풀이
public int kthSmallest(int[][] matrix, int k) {
    int childLength = matrix[0].length;
    int[] tmp = new int[matrix.length * childLength];
    int count = 0;

    for(int i = 0; i < matrix.length; i++) {
        for(int m : matrix[i]) {
            tmp[count++] = m;
        }
    }

    Arrays.sort(tmp);
    return tmp[k - 1];
}
결과

 

문제는 2중 배열에서 k 번째 작은 값을 구하는 문제였습니다.

처음에는 앞에서부터 k + 1 정도의 인자만 확인해서 k 번째 값을 구하려고 했습니다.

그러다 중간중간 나오는 에러들을 보고 문제를 잘못 이해했다 싶어 다시 문제를 풀었네요.

반응형