n个最小和

最新推荐文章于 2021-10-29 19:41:31 发布

数学作曲家

最新推荐文章于 2021-10-29 19:41:31 发布

阅读量395

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/qq_42403069/article/details/86630211

本文介绍了一种算法，用于解决从两个数组中选择元素相加得到的n^2个和中，找到最小的n个和的问题。输入包括两个数组A和B，每个数组包含n个整数，通过两两组合并求和，最后输出从小到大的最小n个和。算法使用了优先队列和结构体来优化搜索过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目
问题描述
给出两个包含 n 个整数的数组 A，B。分别在 A, B 中任意出一个数并且相加，可以得到 n^2个和。求这些和中最小的 n 个。
输入格式
输入第一行一个整数n(1≤n≤50000)。接下来一行输入数组 A，用空格隔开。接下来一行输入数组 B，用空格隔开。1<=ai<=10^9.
输出格式
从小到大输出最小的 n 个和，用空格隔开。
样例输入
4
1 3 5 7
2 4 6 8
样例输出
3 5 5 7
代码

#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
int a[100005],b[100005],c[100005];int n;;
struct data{
    int aa,bb,c;
}o[100005],z;
bool operator<(data x,data y)
{  return x.c>y.c;
}
priority_queue<data> q;
int main()
{  
   cin>>n;int l1=1,l2=1;
   for (int i=1;i<=n;++i)
   {  cin>>a[i];
   }
   sort(a+1,a+1+n);
   for (int i=1;i<=n;++i)
   {  cin>>b[i];
   }
   sort(b+1,b+1+n);
   for (int i=1;i<=n;++i)
   {  o[i].aa=1;
      o[i].bb=i;
      o[i].c=a[o[i].aa]+b[o[i].bb];
      q.push(o[i]);
   }
   for (int i=1;i<=n;++i)
   {  
      z=q.top();
      q.pop();
      cout<<z.c;
      if(i!=n)cout<<" ";
      z.aa++;
      z.c=a[z.aa]+b[z.bb];
      q.push(z);     
   }
}