LeetCode 剑指 Offer II 061. 和最小的 k 个数对***

具体思路:

没有看出来是最小堆问题,光想着双指针怎么调整了。

具体代码:

class Solution {
public:
    vector<vector<int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
        int m=nums1.size();
        int n=nums2.size();
        auto cmp=[](pair<int,int>& pa,pair<int,int>& pb){
            return pa.first+pa.second<pb.first+pb.second;
        };
        priority_queue<pair<int,int>, vector<pair<int,int>>, decltype(cmp)>pri(cmp);
        for(int i=0;i<k&&i<m;i++){
            for(int j=0;j<k&&j<n;j++){
                if(pri.size()<k){
                    pri.push({nums1[i],nums2[j]});
                }else if(pri.top().first+pri.top().second>nums1[i]+nums2[j]){
                    pri.pop();
                    pri.push({nums1[i],nums2[j]});
                }
            }
        }
        vector<vector<int>>ret;
        while(!pri.empty()){
            k--;
            ret.insert(ret.begin(),{pri.top().first,pri.top().second});
            cout<<ret[0][0]<<" "<<ret[0][1]<<endl;
            pri.pop();
        }
        return ret;
    }
};
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值