从m个数组(每个数组中n个元素)中各任取一个元素求和,得到n^m个元素。从小到大输出前k个合并后的元素。
常规做法:将所有元素求和排序,O((n^m)*log(n^m)),超时
巧妙做法:
1.分别将m个数组进行排序,得到m个有序表 O(m*n*logn)
2.每个有序表中各取一个元素相加后排序,求前k小的元素
简化:m=2时,设两个有序表为a[0,1...n-1],b[0,1...n-1],从a和b数组中各取一个元素,求和得到n^2个数,求前k小元素
构造新的有序表,第i个有序表元素为:b[i]+a[j],1<=j<=n-1
则问题转化为归并n个有序表(每个表有n个元素),时间复杂度为O(n*n*logn),如果只求前k小元素,则时间复杂度降为O(k*logn)
m>2时,可以每两个数组构造出n个有序表,得到前k小元素,再依次和其他数组进行相同处理,总时间复杂度:O(m*k*logn)
时间复杂度下降的原因是只考虑了前k小元素,而未将后面的元素有序化。
注:合并n个有序表,共含m个元素,时间复杂度:O(m*logn)
原理:有序表内部已经有序,因此同时只需要维护有序表大小(n)个数的大小关系,在取出这n个数中的最小值后,再将此有序表中的后一个元素添加到此维护序列中
#include<cstdio>
#include<queue>
#include<algorithm>
using namespace std;
struct Item{
int s,b;//s=A[a]+B[b]
Item(int s,int b):s(s),b(b){ }
bool operator<(const Item& rhs) const{
return s>rhs.s;
}
};
void merge(int* A,int* B,int* C,int n){
priority_queue<Item> q;
for(int i=0;i<n;i++)
q.push(Item(A[i]+B[0],0));
for(int i=0;i<n;i++){
Item item=q.top();q.pop();//取出A[a]+B[b]
C[i]=item.s;//合并值的前n个值依次加入
int b=item.b;
if(b+1<n) q.push(Item(item.s-B[b]+B[b+1],b+1));//更新s值加入合并表
}
}
const int maxn=768;
int A[maxn][maxn];
int main(){
int n;
//freopen("a.txt","r",stdin);
while(scanf("%d",&n)!=EOF){
//预处理,使数组有序
for(int i=0;i<n;i++){
for(int j=0;j<n;j++) scanf("%d",&A[i][j]);
sort(A[i],A[i]+n);
}
//两两合并:构造n个有序表,再将其合并
for(int i=1;i<n;i++)
merge(A[0],A[i],A[0],n);
//输出结果
printf("%d",A[0][0]);
for(int i=1;i<n;i++) printf(" %d",A[0][i]);
printf("\n");
}
}