CODEVS 1245 最小的N个和

题目描述 Description

有两个长度为 N 的序列 A 和 B，在 A 和 B 中各任取一个数可以得到 N^2 个和，求这N^2 个和中最小的 N个。

输入描述 Input Description

第一行输入一个正整数N；第二行N个整数Ai 且Ai≤10^9；第三行N个整数Bi，
且Bi≤10^9

输出描述 Output Description

输出仅一行，包含 n 个整数，从小到大输出这 N个最小的和，相邻数字之间用
空格隔开。

样例输入 Sample Input

5
1 3 2 4 5
6 3 4 1 7

样例输出 Sample Output

2 3 4 4 5

数据范围及提示 Data Size & Hint

【数据规模】 对于 100%的数据，满足 1≤N≤100000。

分析

将两个队列分为一个主一个次。排序次队列。
堆维护主队列中每个元素目前达到的大小。主元素每次被调用大小都变为它与下个次元素的和。

代码

#include <bits/stdc++.h>
using namespace std;
int l[100001],r[100001],n,i;
set <pair<int,pair<int,int> > > w;
pair<int,pair<int,int> > t;
int main()
{
    cin>>n;
    for (i = 0; i < n; i++) 
        cin>>l[i];
    for (i = 0; i < n; i++) 
        cin>>r[i];
    sort(r, r + n);
    for (i = 0; i < n; i++) 
        w.insert(make_pair(l[i] + r[0], make_pair(i,0)));
    for (i = 0; i < n; i++)
    {
        t = *w.begin();
        w.erase(w.begin());
        cout<<t.first<<' ';
        w.insert(make_pair(l[t.second.first] + r[t.second.second + 1],
                           make_pair(t.second.first, t.second.second + 1)));
    }
}