码迷,mamicode.com
首页 > 其他好文 > 详细

find-k-pairs-with-smallest-sums

时间:2016-07-10 15:27:54      阅读:145      评论:0      收藏:0      [点我收藏+]

标签:

// https://discuss.leetcode.com/topic/50527/java-10ms-solution-no-priority-queue

class Solution {
public:
    vector<pair<int, int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());

        vector<int> buf1(nums1.size(), 0);
        vector<pair<int, int>> vpair;
        int minV = INT_MAX;
        int tmpV = -1;
        
        while (vpair.size() < k) {
            for (int i=0; i<nums1.size(); ++i) {
                if (buf1[i] < nums2.size() && nums1[i] + nums2[buf1[i]] < minV) {
                    minV = nums1[i] + nums2[buf1[i]];
                    tmpV = i;
                }
            }
            
            if (tmpV != -1) {
                vpair.push_back(make_pair(nums1[tmpV], nums2[buf1[tmpV]]));
                buf1[tmpV] = buf1[tmpV] + 1;
                minV = INT_MAX;
                tmpV = -1;
            }
            else {
                break;
            }
            
        }
        
        return vpair;
    }
};

 

find-k-pairs-with-smallest-sums

标签:

原文地址:http://www.cnblogs.com/charlesblc/p/5657705.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!