leetcode 373. Find K Pairs with Smallest Sums

思路: 
1. 利用数组mark记录nums1中每个元素[i]当前与nums2中关联的位置[ji],
初始关联状态为[0, 0], [1, 0], … , [i, 0], …, [nums1.size()-1,0]; 
2. 利用小顶堆比较这nums1.size()个状态对应的值( nums1[i]+nums2[ji] ),找出当前的最小值。
更新mark[min]的值,并将(nums1[min]+nums2[mark[min]])值加入堆中。

class Solution {
public:
    vector<pair<int, int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
        if (nums1.size() == 0 || nums2.size()  == 0 ) 
            return retVec;
        for (vector<int>::size_type i = 0; i < nums1.size(); ++i){
                mark.push_back(0);
                que.push(new Node(nums1[i]+nums2[0],i));
        }
        for (int i = 0; i < k; ++i){
            if (que.empty() == false){
                int smallestSum = que.top()->sum;
                unsigned int pos = que.top()->pos;
                que.pop();
                int value2=nums2[mark[pos]];
                retVec.push_back(make_pair(smallestSum-value2,value2));
                mark[pos]++;// 将nums1[pos]关联的nums2中元素位置更新(向后移动一格)
                if(mark[pos] >= nums2.size()) 
                    continue;
                else{
                    int sum=nums1[pos]+nums2[mark[pos]];
                    que.push(new Node(sum,pos));
                }
            }
        }
        return retVec;
    }
private:
   class Node{
      public:
      int sum;// 位于不同数组的两个元素之和
      unsigned int pos;// 其中一个元素在nums1中的索引
      Node(int s,unsigned int p):sum(s),pos(p){}
    };
    struct cmp{//用于构建小顶堆
        bool operator()(Node* a,Node* b){
            return a->sum > b->sum;// 是'>',不是'<'
        }
    };
    vector<int> mark;
    priority_queue<Node*,vector<Node*>,cmp> que;
    vector<pair<int,int>> retVec; 
};


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值