資訊專欄INFORMATION COLUMN
摘要：先放一行，或一列把堆頂的最小元素取出來，取次，如果該有下一行下一列的，放入堆中最小的個元素已經在上面的循環被完了，下一個堆頂元素就是
Problem
Given a n x n matrix where each of the rows and columns are sorted in ascending order, find the kth smallest element in the matrix.
Note that it is the kth smallest element in the sorted order, not the kth distinct element.
Example:
matrix = [ [ 1, 5, 9], [10, 11, 13], [12, 13, 15] ], k = 8, return 13.
Note:
You may assume k is always valid, 1 ≤ k ≤ n^2.
class Solution {
public int kthSmallest(int[][] matrix, int k) {
int n = matrix.length;
Queue queue = new PriorityQueue((a, b)->a.val-b.val);
//先放一行，或一列
for (int i = 0; i < n; i++) queue.offer(new Node(i, 0, matrix[i][0]));
//把堆頂的最小元素取出來，取k-1次，如果該node有下一行/下一列的node，放入堆中
for (int i = 0; i < k-1; i++) {
Node node = queue.poll();
if (node.y == n-1) continue;
else queue.offer(new Node(node.x, node.y+1, matrix[node.x][node.y+1]));
}
//最小的k-1個元素已經在上面的for循環被poll完了，下一個堆頂元素就是kth smallest
return queue.poll().val;
}
}
class Node {
int x;
int y;
int val;
public Node(int x, int y, int val) {
this.x = x;
this.y = y;
this.val = val;
}
}
文章版權歸作者所有，未經允許請勿轉載,若此文章存在違規行為，您可以聯系管理員刪除。
轉載請注明本文地址：http://specialneedsforspecialkids.com/yun/72219.html
摘要：復雜度是，其中。這做法和異曲同工。看了網上給的解法，沒有二分，二分的是結果。每次找到一個，然后求比它小的元素的個數，根據個數大于還是小于來二分。參考算的時候可以優化 378. Kth Smallest Element in a Sorted Matrix 題目鏈接：https://leetcode.com/problems... 求矩陣里面第k小的數，首先比較容易想到的是用heap來做...
摘要：因此我們可以采用部分元素堆排序即可。即我們每次只需要可能構成第個元素的值進行堆排序就可以了。 題目要求 Given a n x n matrix where each of the rows and columns are sorted in ascending order, find the kth smallest element in the matrix. Note that...
Given a n x n matrix where each of the rows and columns are sorted in ascending order, find the kth smallest element in the matrix.Note that it is the kth smallest element in the sorted order, not the...
摘要：中序遍歷復雜度時間空間思路因為左節點小于根節點小于右節點，二叉搜索樹的一個特性就是中序遍歷的結果就是樹內節點從小到大順序輸出的結果。這里采用迭代形式，我們就可以在找到第小節點時馬上退出。這樣我們就可以用二叉樹搜索的方法來解決這個問題了。 Kth Smallest Element in a BST Given a binary search tree, write a function...
摘要：解題思路本題需要找的是第小的節點值，而二叉搜索樹的中序遍歷正好是數值從小到大排序的，那么這題就和中序遍歷一個情況。 Kth Smallest Element in a BSTGiven a binary search tree, write a function kthSmallest to find the kth smallest element in it. Note: You ...
閱讀 929·2023-04-26 01:34
閱讀 3363·2023-04-25 20:58
閱讀 3289·2021-11-08 13:22
閱讀 2118·2019-08-30 14:17
閱讀 2526·2019-08-29 15:27
閱讀 2679·2019-08-29 12:45
閱讀 3003·2019-08-29 12:26
閱讀 2816·2019-08-28 17:51
