給定一個 n x n 矩陣,其中每行和每列元素均按升序排序,找到矩陣中第k小的元素。
請注意,它是排序後的第k小元素,而不是第k個元素。示例:
matrix = [ [ 1, 5, 9], [10, 11, 13], [12, 13, 15] ], k = 8, 返回 13。說明:
你可以假設 k 的值永遠是有效的, 1 ≤ k ≤ n2 。二分 O(nlgX) X是最大值和最下值得差值。我們每次二分最大值和最小值的和。找出中間值,來確認中間值是第幾大的數字,然後和K比較。
class Solution { public: int kthSmallest(vector<vector<int>>& matrix, int k) { int left = matrix[0][0], right = matrix.back().back(); while (left < right) { int mid = left + (right - left) / 2; int cnt = search(matrix, mid); if (cnt < k) left = mid + 1; else right = mid; } return left; } int search(vector<vector<int>>& matrix, int target) { int n = matrix.size(), i = n - 1, j = 0, ans = 0; while (i >= 0 && j < n) { if (matrix[i][j] <= target) { ans += i + 1; //如果ma[i][j]<=target,則說明第j列上面的數字都小於target,所以ans+= i+1 ++j; } else { --i; } } return ans; } };
Leetcode 378. 有序矩陣中第K小的元素
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.