Leetcode_find_k_pairs_smallest_sum

博客提到利用堆来维护当前最小值,还表示要学习他人代码风格,涉及数据结构中堆的应用及代码学习方面的信息技术内容。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

利用堆,维护当前最小值。

vector<vector<int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
        auto cmp = [](pair<int, int> x, pair<int, int> y){
            return x.first + x.second < y.first + y.second;};
        priority_queue<pair<int, int>, vector<pair<int, int>>, decltype(cmp)> heap(cmp);
        vector<vector<int>> result;
        for(int i = 0; i < nums1.size() && i < k; i++){
            for(int j=0; j < nums2.size() && j < k; j++){
                auto tmp = make_pair(nums1[i], nums2[j]);
                if(heap.size() < k){
                    heap.push(tmp);
                } else {
                    if(!cmp(heap.top(), tmp)){
                        heap.pop();
                        heap.push(tmp);
                    }
                }
            }
        }
        
        while(!heap.empty()){
            result.push_back({heap.top().first, heap.top().second});
            heap.pop();
        }
        return result;
        
    }

一个别人的写法,学习一下代码风格

 

vector<vector<int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
        if (nums1.empty() || nums2.empty())
            return vector< vector<int> >();
        
        using Pair = pair<int, int>;
        
        auto cmp_func = [&](Pair a, Pair b){ 
            return (nums1[a.first] + nums2[a.second]) > 
                (nums1[b.first] + nums2[b.second]);
        };
        
        using Pri = priority_queue<Pair, vector<Pair>,  decltype(cmp_func)>;
        
        Pri minheap(cmp_func);
        
        vector< vector<int> > result;
        
        for (int i = 0; i < nums1.size() && i < k; i++)
        {
            minheap.push({i, 0});
        }
        
        while (result.size() < k && !minheap.empty())
        {
            auto min = minheap.top(); 
            minheap.pop();
            result.push_back({nums1[min.first], nums2[min.second]});
            
            if (min.second < nums2.size() - 1)
            {
                minheap.push({min.first, min.second + 1});
            }
        }
        
        return result;
    }

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值