有序矩陣中第K小的元素Scala實現

2021-10-10 03:18:23 字數 901 閱讀 9484

運用scala中的語法實現

有序矩陣中第k小的元素

給定乙個 n x n 矩陣,其中每行和每列元素均按公升序排序,找到矩陣中第 k 小的元素。

請注意,它是排序後的第 k 小元素,而不是第 k 個不同的元素。

示例:matrix = [

[ 1, 5, 9],

[10, 11, 13],

[12, 13, 15]

],k = 8,

返回 13。

**如下(示例):

這就是二分法的思路。假定查詢的範圍為[left, right],首先計算int mid = left + (right - left) / 2,

然後在矩陣中計數有多少個元素小於等於mid,這個數量為count。

如果count < k,那麼第k小的數至少為mid + 1,所以left = mid + 1。

反之,right = mid。

迴圈結束的條件為left >= right,此時left即為答案。

def kthsmallest(matrix: array[array[int]], k: int): int =

left

}

def countnotmorethanmid(matrix: array[array[int]], mid: int, len: int)=else

count

}

例如 mid=8:

我們可以看到,矩陣中大於 midmid 的數就和不大於 midmid 的數分別形成了兩個板塊,沿著一條鋸齒線將這個矩形分開。其中左上角板塊的大小即為矩陣中不大於 midmid 的數的數量。

有序矩陣中第k小元素

題目 給定乙個 n x n 矩陣,其中每行和每列元素均按公升序排序,找到矩陣中第 k 小的元素。請注意,它是排序後的第 k 小元素,而不是第 k 個不同的元素。看到有序就會想到二分查詢,而本題的二分查詢十分的有趣。根據這個矩陣的定義,我們知道,最小的元素是最左上角元素,最大的元素是最左下角元素。由此...

有序矩陣中第K小的元素

給定乙個 n x n 矩陣,其中每行和每列元素均按公升序排序,找到矩陣中第k小的元素。請注意,它是排序後的第k小元素,而不是第k個元素。示例 matrix 1,5,9 10,11,13 12,13,15 k 8,返回 13。說明 你可以假設 k 的值永遠是有效的,1 k n2 分析 因為是整數,可使...

LeetCode 有序矩陣中第K小的元素

給定乙個 n x n 矩陣,其中每行和每列元素均按公升序排序,找到矩陣中第k小的元素。請注意,它是排序後的第k小元素,而不是第k個元素。示例 matrix 1,5,9 10,11,13 12,13,15 k 8,返回 13。public boolean guess int matrix,int mi...