leetcode 每日一题1493. 有序矩阵中的第k个最小数组和

题目描述跳转到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:感觉有点动态规划的意思，又不确定，动态规划就是将大问题分解成小问题逐一解决，这段代码的解法也是先求出小范围的和最后在扩展到所有的数组。