【转】序列合并(洛谷P1631)

【转】序列合并(洛谷P1631)

序列合并

题目描述

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

输入格式

第一行一个正整数N；

第二行N个整数Ai, 满足Ai<Ai+1 ，且Ai<10^9;

第三行N个整数Bi, 满足Bi<Bi+1 ，且Bi<10^9;

【数据规模】

对于50%的数据中，满足1<=N<=1000；

对于100%的数据中，满足1<=N<=100000。

---

这道题的解法很难想到；首先把A序列和B序列都排好序（从小到大），然后把b[1]+a[i]（ i 从1到n）全部入队，还要把i入队（就是a的坐标也入队），显然要用到结构体，STL还有个更加方便的东西pair，作用相当于结构体；然后出n次队，每次出来的数就是最小的和，出来的同时，在把出队那个数的下标后面那一位进队；

双倍经验

代码：

#include<bits/stdc++.h>
#define LL long long
using namespace std;
int a[100100];
int b[100100];
priority_queue< pair<int,int>,vector< pair<int,int> >,greater< pair<int,int> > >qu;
int to[100100];
int main(){
    
    
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int i=1;i<=n;i++) scanf("%d",&b[i]);
	for(int i=1;i<=n;i++) to[i]=1;
	for(int i=1;i<=n;i++) qu.push(pair<int,int>(a[i]+b[1],i));
	while(n--){
    
    
		printf("%d ",qu.top().first);
		int s=qu.top().second;
		qu.pop();
		qu.push(pair<int,int>(a[s]+b[++to[s]],s));
	}
	return 0;
}