Leetcode 373. 查找和最小的K对数字解题思路及C++实现

最新推荐文章于 2022-09-30 08:46:28 发布

PaniniGu

最新推荐文章于 2022-09-30 08:46:28 发布

阅读量658

点赞数

分类专栏： Leetcode

本文链接：https://blog.csdn.net/gjh13/article/details/90273634

版权

本文详细介绍了如何解决LeetCode的373题，通过使用最大堆来寻找和最小的K对数字。文章中提到，可以利用C++的priority_queue配合自定义比较函数cmp实现，重点在于只访问两个数组的前k个元素，降低了时间复杂度至O(k*k)。

摘要由CSDN通过智能技术生成

解题思路：

使用最大堆来存储k个和最小的数组组合，然后每进来一个数组组合，只需要和最大堆的堆顶进行比较即可。

用C++的priority_queue实现最大堆时，需要用一下pair来组合来自nums1和nums2数组的数，然后自己写一个比较结构体cmp，比较数组和的大小。

在循环中的一个trick在于，限制为访问nums1和nums2的前k个数即可。所以那一部分的时间复杂度为O(k*k)。

class Solution {
public:
    vector<vector<int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
        if(k == 0) return {
  {}};
        priority_queue<pair<int, int>, vector<pair<int, int>>, cmp> pq;   //最大堆
        for(int i = 0; i < nums1.size() && i < k; i++){
            for(int j = 0; j < nums2.size() && j < k; j++){
                pq.push(make_pair(nums1[i], nums2[j]));
                if(pq.size() > k) pq.pop();
            }