373. 查找和最小的 K 对数字

题目描述:373. 查找和最小的 K 对数字

题解

直观的思路是每一个结点都向右侧和下侧扩展,但是这种方案会引起重复。

修正后的方案是将第一列结点添加入heap中后,每次取数据后只向右侧进行扩展。

cpp 复制代码
class Solution {
public:
    struct node
    {
        int val;
        int x, y;

        // 取巧定义比较函数
        bool operator < (const node& other) const
        {
            return val > other.val;
        }
    };

    node Add(int val, int x, int y)
    {
        return node{ val, x, y };
    }

    // K对最小数组
    vector<vector<int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) 
    {
        // 默认为大顶堆,但是定义特殊比较函数,因此为小顶堆
        priority_queue<node> q;
        vector<vector<int>> ans;
        int n, m;
        n = nums1.size(); m = nums2.size();
        for (int i = 0; i < min(k, n); i++)
        {
            q.push(Add(nums1[i]+nums2[0], i, 0));
        }
        while (!q.empty() && k)
        {
            node tmp = q.top(); q.pop(); 
            k--;
            int x, y;
            x = tmp.x, y = tmp.y;
            ans.push_back({nums1[x], nums2[y]});
            // 只进行向右侧进行扩展
            if(y < m-1)
                q.push(Add(nums1[x] + nums2[y + 1], x, y + 1));
        }
        return ans;
    }
};
相关推荐
ID_1800790547324 分钟前
淘宝实时拍立淘按图搜索数据|商品详情|数据分析提取教程
算法·数据分析·图搜索算法
l1t25 分钟前
Lua与LuaJIT的安装与使用
算法·junit·单元测试·lua·luajit
Emilia486.2 小时前
【Leetcode&nowcode】代码强化练习(二叉树)
算法·leetcode·职场和发展
墨染点香2 小时前
LeetCode 刷题【135. 分发糖果】
算法·leetcode·职场和发展
秋风战士2 小时前
通信算法之336 :3GPPMixed Mode Turbo Decoder
算法·matlab·fpga开发·信息与通信·基带工程
是那盏灯塔2 小时前
【算法】——动态规划之01背包问题
数据结构·c++·算法·动态规划
im_AMBER2 小时前
Leetcode 41
笔记·学习·算法·leetcode
jinmo_C++3 小时前
数据结构_深入理解堆(大根堆 小根堆)与优先队列:从理论到手撕实现
java·数据结构·算法
IT19953 小时前
OpenSSL3.5.2实现SM3数据摘要生成
算法·哈希算法·散列表
Excuse_lighttime3 小时前
排序数组(快速排序算法)
java·数据结构·算法·leetcode·eclipse·排序算法