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;
}