51nod 1821 最优集合

题意

一个集合S的优美值定义为：最大的x，满足对于任意i∈[1,x]，都存在一个S的子集S’，使得S’中元素之和为i。
给定n个集合，对于每一次询问，指定一个集合S1和一个集合S2，以及一个数k，要求选择一个S2的子集S3(|S3|<=k)，使得S1∪S3的优美值最大。
（集合元素可以重复）
n，集合大小<=1000，元素大小<=10^9，T<=10000

分析

考虑如何求出一个集合的优美值。我们可以把数从小到大加入，如果当前数x<=当前优美值+1，则优美值可以增大x，反之则这就是最终答案。
那么我们可以对S2维护一个双向链表，每次当S1不能加入数的时候，就从S2中取一个最大且可以被假如的数加入即可。

一开始超时啦，后来把读入优化加上就A了。。。

代码

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<algorithm>using namespace std;const int N=1005;int n,tot[N],a[N][N],r[N],l[N];int read(){    int x=0,f=1;char ch=getchar();    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}    return x*f;}int main(){    n=read();    for (int i=1;i<=n;i++)    {        tot[i]=read();        for (int j=1;j<=tot[i];j++) a[i][j]=read();        sort(a[i]+1,a[i]+tot[i]+1);    }    int T=read();    while (T--)    {        int x=read(),y=read(),k=read();        int pts=0,ret=0,ans=0;        for (int i=1;i<=tot[y];i++) r[i]=i+1,l[i]=i-1;        r[0]=1;l[tot[y]+1]=tot[y];        for (int i=1;i<=tot[x];i++)            if (a[x][i]<=ans+1) ans+=a[x][i];            else            {                if (ret==k) break;                while (r[pts]<=tot[y]&&a[y][r[pts]]<=ans+1) pts++;                if (pts) ans+=a[y][pts],ret++,l[r[pts]]=l[pts],r[l[pts]]=r[pts],pts=l[pts],i--;                else break;            }        if (ret<k)        {            for (int i=1;i<=k-ret;i++)            {                while (r[pts]<=tot[y]&&a[y][r[pts]]<=ans+1) pts++;                if (pts) ans+=a[y][pts],ret++,l[r[pts]]=l[pts],r[l[pts]]=r[pts],pts=l[pts];                else break;            }        }        printf("%d\n",ans);    }    return 0;}