1. 程式人生 > >Leetcode 378.有序矩陣中第k小的元素

Leetcode 378.有序矩陣中第k小的元素

有序矩陣中第k小的元素

給定一個 n x n 矩陣,其中每行和每列元素均按升序排序,找到矩陣中第k小的元素。
請注意,它是排序後的第k小元素,而不是第k個元素。

示例:

matrix = [

[ 1, 5, 9],

[10, 11, 13],

[12, 13, 15]

],

k = 8,

 

返回 13。

說明:
你可以假設 k 的值永遠是有效的, 1 ≤ k ≤ n

 

根據二分搜尋法,獲取中間值,然後搜尋他是否為第k個值。

 

 1 class Solution {
2 public int kthSmallest(int[][] matrix, int k) { 3 int len = matrix.length; 4 int low = matrix[0][0],high= matrix[len-1][len-1]; 5 while(low<=high){ 6 int mid = low + (high-low)/2; 7 int count = helper(matrix,mid); 8 if(count<k) low = mid+1;
9 else high = mid-1; //排除mid不在矩陣內的情況,所以只能等到low和high時才退出迴圈 10 } 11 return low; 12 } 13 private static int helper(int[][] matrix, int mid) { 14 int i = matrix.length-1,j=0; 15 int res = 0; 16 while(i>=0&&j<matrix[0].length){ 17 if
(matrix[i][j]>mid) i--; 18 else{ 19 res+=i+1; 20 j++; 21 } 22 } 23 return res; 24 } 25 }