[Cogs727] [网络流24题#2] 太空飞行计划 [网络流，最小割]

建图：源点—(w[i])—>实验—(∞)—>仪器—(cost[i])—>汇点，

如果该实验造成收益，则仪器到汇点的边在最小割中，

如果该实验造成损失，则源点到实验的边在最小割中，

故答案就是所有实验的所有收益（不考虑仪器花费的）减去最小割。

特别注意，最后统计方案时，不能直接判断边权是否为0，因为当花费等于收益时，这种方法会将该实验计入答案而题目则不允许，故应该在最后Bfs出的level数组中判断是否可达，再通过实验把所需仪器统计出来。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <ctime>
#include <algorithm>
#include <queue>
#include <vector>

using namespace std;

template<const int _n>
struct Edge
{
    struct Edge_base { int    to,next,w; }e[_n];
    int    p[_n],cnt;
    Edge()    { clear(); }
    inline int    start(const int x) { return p[x]; }
    inline void    insert(const int x,const int y,const int z)
    { e[++cnt].to=y; e[cnt].next=p[x]; e[cnt].w=z; p[x]=cnt; return ; }
    inline Edge_base&    operator[](const int x) { return e[x]; }
    inline void    clear() { cnt=1,memset(p,0,sizeof(p)); }
};

int    n,m,SSS,TTT;
int    w[210],cost[210];
int    level[210],cur[210];
char    str[11000];
Edge<25000>    e;
vector<pair<int,int> >    vec;

bool    Bfs(const int S)
{
    int    i,t;
    queue<int>    Q;
    memset(level,0,sizeof(int)*(n+m+3));
    level[S]=1;Q.push(S);
    while(!Q.empty())
    {
        t=Q.front(),Q.pop();
        for(i=e.start(t);i;i=e[i].next)
        {
            if(!level[e[i].to] && e[i].w)
            {
                level[e[i].to]=level[t]+1;
                Q.push(e[i].to);
            }
        }
    }
    return    level[TTT];
}

int    Dfs(const int S,const int bk)
{
    if(S==TTT)return bk;
    int    rest=bk;
    for(int &i=cur[S];i;i=e[i].next)
    {
        if(level[e[i].to]==level[S]+1 && e[i].w)
        {
            int    flow=Dfs(e[i].to,min(rest,e[i].w));
            e[i].w-=flow;
            e[i^1].w+=flow;
            if((rest-=flow)<=0)break;
        }
    }

    if(rest==bk)level[S]=0;
    return bk-rest;
}

int    Dinic()
{
    int    i,j,flow=0;
    int    Ans[210]={0};
    while(Bfs(SSS))
    {
        memcpy(cur,e.p,sizeof(int)*(n+m+3));
        flow+=Dfs(SSS,0x3f3f3f3f);
    }

    for(i=1;i<=n;++i)
    {
        if(level[i])
        {
            printf("%d ",i);
            for(j=e.start(i);j;j=e[j].next)
                Ans[e[j].to-n]=1;
        }
    }
    printf("\n");
    for(i=1;i<=m;++i)
    {
        if(Ans[i])printf("%d ",i);
    }
    printf("\n");

    return flow;
}

void    Build()
{
    int    i;
    SSS=n+m+1,TTT=n+m+2;
    for(i=1;i<=n;++i)
    {
        e.insert(SSS,i,w[i]);
        e.insert(i,SSS,0);
    }
    for(i=0;i<(int)vec.size();++i)
    {
        e.insert(vec[i].first,vec[i].second+n,0x3f3f3f3f);
        e.insert(vec[i].second+n,vec[i].first,0);
    }
    for(i=n+1;i<=m+n;++i)
    {
        e.insert(i,TTT,cost[i-n]);
        e.insert(TTT,i,0);
    }
    return ;
}

#define    EOL    -1
int    getint(int&    data,int & t)
{
    char    ch;data=0;
    ch=str[t++];
    while((ch<'0' || ch>'9') && ch)ch=str[t++];
    while(ch>='0' && ch<='9')data=data*10+ch-48,ch=str[t++];
    if(!data)return EOL;
    return 1;
}

int main()
{
    freopen("shuttle.in","r",stdin);
    freopen("shuttle.out","w",stdout);

    int    i,c,Sum=0;

    scanf("%d%d",&n,&m);
    for(i=1;i<=n;++i)
    {
        scanf("%d",&w[i]);
        Sum+=w[i];
        fgets(str,10000,stdin);
        int t=0;
        while(~getint(c,t))
        {
            vec.push_back(make_pair(i,c));
            if(t==-1)break;
        }
    }
    for(i=1;i<=m;++i)
    {
        scanf("%d",&cost[i]);
    }

    Build();

    printf("%d\n",Sum-Dinic());

    return 0;
}