leetcode + 优先队列，注意语法的使用

最新推荐文章于 2024-08-16 00:32:27 发布

小堃哥

最新推荐文章于 2024-08-16 00:32:27 发布

阅读量623

点赞数

分类专栏： Leetcode

本文链接：https://blog.csdn.net/u013554860/article/details/80211973

版权

Leetcode 专栏收录该内容

450 篇文章 0 订阅

订阅专栏

点击打开链接

class Solution {
public:
    struct cmp{
        bool operator()(pair<int, int>num1, pair<int, int>num2){
            return num1.first +num1.second > num2.first+num2.second;
        }
    };
    vector<pair<int, int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
        vector<pair<int, int>> res;
        priority_queue< pair<int, int>, vector<pair<int, int> >, cmp>  pq;  //最小堆
        int i=0, j=0, m=nums1.size(), n=nums2.size();
        for(i=0; i<m; i++){
            for(j=0; j<n;j++){
                pq.push(make_pair(nums1[i], nums2[j]));
            }
        }
        while (k>0&&pq.size()>0) {
            res.push_back(pq.top());
            pq.pop();
            k-=1;
        }
        return res;
    }
};