首页 > 其他 > 详细

find-k-pairs-with-smallest-sums

时间:2016-07-10 15:25:34      阅读:305      评论: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
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!