Algorithm
[Algorithm][Recursion] 문제 풀이 #22 - Kth Smallest Element in a Sorted Matrix
Lee David
2022. 8. 13. 08:11
반응형
문제 링크
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 번째 값을 구하려고 했습니다.
그러다 중간중간 나오는 에러들을 보고 문제를 잘못 이해했다 싶어 다시 문제를 풀었네요.
반응형