题面
背景
输入
输出
翻译(渣自翻)
给定K个包含K个数字的表,要求将其能产生的\( k^{k} \)个值中最小的K个输出出来
题解
k路归并问题的经典问题
可以转化为二路归并问题求解
考虑A[],B[]两个有序数组
使用堆,记录一些二元组\( (x,y) \),x表示值,y表示对应的b的下标,因为我们是把b合并到a上,所以我们能够根据记录的下标推出后面的值
然后两两合并
所以就很简单
放代码
#include <cstdio> #include <algorithm> #include <queue> using namespace std; struct asshole{ int w,t; bool operator < (const asshole b) const{ return w>b.w; } }; priority_queue<asshole> q; int kz[10000],kk[10000]; int n; void merge(int *a,int *b,int *c){ while(!q.empty()) q.pop(); for(int i=1;i<=n;i++) q.push((asshole){a[i]+b[1],1}); for(int i=1;i<=n;i++){ asshole in = q.top(); q.pop(); a[i]=in.w; int pos=in.t; if(pos+1<=n) q.push((asshole){in.w-b[pos]+b[pos+1],pos+1}); } } int main(){ while(scanf("%d",&n)==1){ for(int i=1;i<=n;i++) scanf("%d",&kz[i]); sort(kz+1,kz+n+1); for(int j=1;j<=n-1;j++){ for(int i=1;i<=n;i++) scanf("%d",&kk[i]); sort(kk+1,kk+n+1); merge(kz,kk,kz); } for(int i=1;i<=n-1;i++) printf("%d ",kz[i]); printf("%d",kz[n]); printf("\n"); } }
原文地址:https://www.cnblogs.com/dreagonm/p/9484130.html
时间: 2024-11-08 19:33:31