【UVA–11997 K Smallest Sums 】

时间:2023-03-08 21:18:41
【UVA–11997 K Smallest Sums 】

·哦,这题要用优先队列?那大米饼就扔一个手写堆上去吧!

·英文题,述大意:

      输入n个长度为n的序列(题中是k,2<=k<=750)。一种结果定义为:从每个序列中都要挑选一个数加起来。挑选的不同种结果含有的元素可以重复,现在你需要求出在所有的nn个结果中,找到其中最小的n个结果,然后按照从小到大顺序输出这n个结果。

·分析:

     我们可以从简单情况加以考虑以得到普遍结论。

     当只有1个序列时,那么就直接排个序就可以了(虽然不在数据范围里)。

     当只有两个序列,也就是挑选出两个数(来自不同序列)的和我们要的结果。首先,我们怎么取得最小的那个和?毫无疑问,就是这个序列两个的最小数的和。那么第二小的数怎么取得?嗯嗯,肯定是这样:

 第二小数=Min(1序列最小数+2序列第二小数,1序列第二小数+2序列最小数)

这样一直思考下去,现在我们知道了第i小的结果,要的到第(i+1)小的结果,就有两种选择加以比较。为了便于我们找到单个序列中第i大,我们给所有序列从小到大排序。

      排序后,我们可以知道一个这样的结论:假设现在选择的第p大的组合是a[i]+b[j](注意,排好序了的),那么第p大肯定不会去考虑a[i]+b[j+1],因为a[i]+b[j]<a[i]+b[j+1]。这句奇怪的话只是想说明一个问题,在a[i]+b[j]都还没有被选为答案时,a[k]+b[t](k>=i,t>=j,且等号不同时成立)肯定不用管(管的意思是拿去进行Min的比较)。

      快速维护大小关系,我们可以使用优先队列,将各式各样的组合塞进去。但是我们把所有的压进去,时间耗费太多(n*n啊!)。所以,使用上文的结论,那么上文在CODE中的意义是,a[i]+b[j]在优先队列中时,a[k]+b[t]无需存在。当我们挑选第k大的结果时,就是队首元素啦。那么接下来怎么维护?我们是用有序表:(注意a,b还是排好序了的)

               a1+b1<=a1+b2<=a1+b3……<=a1+bn

                a2+b1<=a2+b2<=a2+b3……<=a2+bn

     这样做的话,我们维护了a的有序,那么对于每个组合,当a[i]+b[j]出队被选为答案后,我们就立刻将a[i]+b[j+1](前提是j+1<=n)加入队列作为将来可能的答案。到此我们可以推而广之,有n个序列时,我们输入一个b就和合并一次,将最小的答案直接塞到a中,操作n-1次合并,就完事啦。

     手写了一个小堆堆,但是这道题数据小,手写堆没发挥优势。

     代码来了:

 #include<stdio.h>
#include<algorithm>
#define Exchange(a,b) a^=b^=a^=b
#define go(i,a,b) for(int i=a;i<=b;i++)
using namespace std;const int N=;
int n,a[N],b[N],val[N],I[N];
struct Heap
{
int sz,cur[N],fa,v;
inline void Up_Adjust(int u)
{
fa=u>>;while(u!=&&val[cur[fa]]>val[cur[u]])
Exchange(cur[fa],cur[u]),fa=(u=fa)>>;
}
inline void Down_Adjust(int u)
{
v=u<<;while(v<=sz){v+=val[cur[v]]>val[cur[v+]]&&v<sz;
if(val[cur[v]]>=val[cur[u]])return;
Exchange(cur[u],cur[v]);v=(u=v)<<;}
}
inline void Insert(int i){cur[++sz]=i,Up_Adjust(sz);}
inline void Delete(){Exchange(cur[],cur[sz]);sz--;Down_Adjust();}
}q;
int main()
{
while(~scanf("%d",&n))
{
go(i,,n)scanf("%d",&a[i]);sort(a+,a+n+);
go(k,,n)
{
go(i,,n)scanf("%d",&b[i]);sort(b+,b+n+);q.sz=;
go(i,,n)val[i]=a[i]+b[],I[i]=,q.Insert(i);
go(j,,n){int i=q.cur[];a[j]=val[i];q.Delete();
if(I[i]<n)val[i]+=-b[I[i]]+b[I[i]+],I[i]++,q.Insert(i);}
}
printf("%d",a[]);go(i,,n)printf(" %d",a[i]);puts("");
}
return ;
}//Paul_Guderian

这是一段很长很长的旅程,用尽所有的时光永无止境

我不停地奔跑呼喊和追寻,在我的路上寻找生命的意义。————汪峰《我的路》