有序矩阵中第K小的元素(通俗解法)

时光怂恿深爱的人放手 提交于 2020-08-04 22:15:20

给定一个 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;
	}
   
}

官方给的二分太难想了。。。

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!