题目描述跳转到leetcode查看
给你一个 m * n 的矩阵 mat,以及一个整数 k ,矩阵中的每一行都以非递减的顺序排列。
你可以从每一行中选出 1 个元素形成一个数组。返回所有可能数组中的第 k 个 最小 数组和。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/find-the-kth-smallest-sum-of-a-matrix-with-sorted-rows
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题
刚开始真的毫无头绪,看了题解说是要用到堆,所以今天学习了堆的相关知识才开始下手。
我的思路是计算前n行的每种情况的和存入大顶堆queue中,最后取出queue中堆顶位置的值。
class Solution {
public int kthSmallest(int[][] mat, int k) {
PriorityQueue<Integer> queue = new PriorityQueue<>((a, b) -> b-a);
queue.offer(0);
int m = mat.length, n = mat[0].length;
for(int i = 0; i< m;i++){
// 截止至当前行的所有组合 各自的和
PriorityQueue<Integer> temp = new PriorityQueue<>((a, b) -> b-a);
for(int j = 0; j< n;j++){
// 第一行直接放进去,以后的每一行的每一个值和temp的每一个值进行加法运算
for(int q: queue){
int tmpValue = q+mat[i][j];
if (temp.size() < k || tmpValue < temp.peek()) {
temp.offer(tmpValue);
// 方便后边取出第k个值,否则后续需要遍历queue到第k个才可取出
if(temp.size() > k){
temp.poll();
}
}
}
}
queue = temp;
}
return queue.peek();
}
}
解决!今天完工啦!
ps:感觉有点动态规划的意思,又不确定,动态规划就是将大问题分解成小问题逐一解决,这段代码的解法也是先求出小范围的和最后在扩展到所有的数组。