题目描述 Description
有两个长度为 N 的序列 A 和 B,在 A 和 B 中各任取一个数可以得到 N^2 个和,求这N^2 个和中最小的 N个。
第一行输入一个正整数N;第二行N个整数Ai 且Ai≤10^9;第三行N个整数Bi, 且Bi≤10^9
输出描述 Output Description
输出仅一行,包含 n 个整数,从小到大输出这 N个最小的和,相邻数字之间用 空格隔开。
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)));
}
}
转载请注明原文地址: https://ju.6miu.com/read-10966.html