题意:
给出两个长度为n的有序表A和B,在A和B中各任取一个元素,可以得到n2个和,求这些和中最小的n个。(不要去重)
数据范围:n<=4e5
解法:
以n=3为例,下表是两两配对的所有情况:
上表中(i,j)格子中的数一定小于等于(i,j+1),因为只有a[i]是不变的只有b[j]进行变化,而b[]是有序的
因为表中每一行都是有序的,因此第1小的数一定出现在第1列里面:
假设第1小的数为(1,1),记录一下答案,现在要求第2小的,如何求出第二小的呢?
可以仿照上面的想法,因为当前最小的一定是每行最左边的那一列里面最小的,现在去除了(1,1)
那么第1行最左边就是(1,2)了,如下图,对这三个数取min就是第2小的数:
不断进行类似操作直到取出n个数即可。
动态取min的操作可以用小根堆实现。
小根堆的每个节点记录{值,横坐标,纵坐标}
总结:
根据表的性质可以发现当前最小一定是表中每行最左边一列的所有数中的最小值
每次取出一个数的时候,就将这个数在表中的右边一个数加入待选队伍
每次取出的最小值就是当前最小值。
code:
#include<bits/stdc++.h>
using namespace std;
const int N=4e5+5;
int a[N],b[N],n;
struct Node{
int v,i,j;
friend bool operator<(Node a,Node b){
return a.v>b.v;
}
};
signed main(){
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]);
priority_queue<Node>q;
for(int i=1;i<=n;i++){
q.push({a[i]+b[1],i,1});
}
for(int i=1;i<=n;i++){
printf("%d\n",q.top().v);
int x=q.top().i,y=q.top().j;
q.pop();
if(y+1<=n){
q.push({a[x]+b[y+1],x,y+1});
}
}
return 0;
}