Leecode 378: 有序矩阵中第K小元素
题目描述: 给定n*n矩阵,其中每行和每列元素均按升序排序,找到矩阵中第k小元素,请注意:是排序后的第k小元素
示例
示例
matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,
返回 13。
解题思路:
- 暴力解题法:将矩阵中所有的数读入一维数组,对这个一维数组进行排序,取第k个值
- 二分解法:
由于是一个矩阵,左上角的数石最小值,右下角的数石最大值(每行和每列按照升序排列)
每次取两个边界的值进行除以2, 得到一个mid,然后根据这个mid统计整个矩阵中小于mid值的个数,如果小于mid的值的个数刚好大于等于k,,那么应该缩小right的值到mid,如果小于k个,那么说明left应该变成mid
暴力算法`
class Solution {
public int kthSmallest(int[][] matrix, int k) {
//暴力解决
if(matrix.length ==0 || matrix[0].length == 0)
return -1;
int rows = matrix.length;
int cols = matrix[0].length;
int number = rows*cols; //统计矩阵中的所有元素个数
int [] list = new int[number];
int count = 0;
for (int i =0; i<rows;i++)
{
for(int j=0;j<cols;j++)
{
list[count] = matrix[i][j];
count++;
}
}
Arrays.sort(list);
return list[k-1];
}
}
二分算法: