給定一個 n x n 矩陣,其中每行和每列元素均按升序排序,找到矩陣中第 k 小的元素。
請注意,它是排序後的第 k 小元素,而不是第 k 個不同的元素。
示例:
matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,
返回 13。
提示:
你可以假設 k 的值永遠是有效的,1 ≤ k ≤ n2 。
來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/kth-smallest-element-in-a-sorted-matrix
著作權歸領釦網絡所有。商業轉載請聯繫官方授權,非商業轉載請註明出處。
//通俗解法
//先降維,二維降維一維,然後用sort進行排序,最後取第k位
public class Main {
public static void main(String[] args) {
int k = 8;
int matrix[][] = {{1, 5, 9},{10, 11, 13},{12, 13, 15}};
System.out.println(kthSmallest(matrix, k));
}
private static int kthSmallest(int[][] matrix, int k) {
int ans = 0;
int t = 0;
int n = matrix.length;
int m = matrix[0].length;
int tmp[] = new int[n * m];
for (int i=0; i<n; i++) {
for (int j=0; j<m; j++) {
tmp[t++] =
matrix[i][j];
}
}
Arrays.sort(tmp);
ans = tmp[k-1];
return ans;
}
}
官方給的二分太難想了。。。