ZOJ 3642 Just Another Information Sharing Problem

自己已经知道的就不要问啦！！！

自己已经知道的就不要问啦！！！

自己已经知道的就不要问啦！！！

自己已经知道的就不要问啦！！！

少加了一条边无限WA，比赛的时候交了31次……

#include <algorithm>
#include <iostream>
#include <fstream>
#include <sstream>
#include <iomanip>

#include <map>
#include <set>
#include <list>
#include <stack>
#include <queue>
#include <deque>
#include <vector>
#include <string>
#include <bitset>
#include <memory>
#include <complex>
#include <numeric>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <time.h>
#include <ctype.h>
#include <locale.h>

using namespace std;

#pragma pack(4)

const double  eps = 1e-8;
const double   pi = acos(-1.0);
const int  inf = 0x7f7f7f7f;

#define   at(a,i) ((a)&(1<<(i)))
#define   nt(a,i) ((a)^(1<<(i)))
#define set1(a,i) ((a)|(1<<(i)))
#define set0(a,i) ((a)&(~(1<<(i))))

#define gret(a,b) (((a)-(b))>eps)
#define less(a,b) (((b)-(a))>eps)
#define greq(a,b) (((a)-(b))>-eps)
#define leeq(a,b) (((b)-(a))>-eps)
#define equl(a,b) (fabs((a)-(b))<eps)

#define lmax(a,b) ((a)>(b)?(a):(b))
#define lmin(a,b) ((a)<(b)?(a):(b))
#define fmax(a,b) (gret(a,b)?(a):(b))
#define fmin(a,b) (less(a,b)?(a):(b))

const int MAXV = 500;
const int MAXE = 50000;

struct node
{
    int v,w;
}G[MAXE];
int _index,pre[MAXV],next[MAXE];

void clear(void)
{
    _index=0;
    memset(pre,-1,sizeof(pre));
}
void add(int u,int v,int w)
{
    G[_index].v=v;
    G[_index].w=w;
    next[_index]=pre[u];
    pre[u]=_index++;

    G[_index].v=u;
    G[_index].w=0;
    next[_index]=pre[v];
    pre[v]=_index++;
}

int dis[MAXV],gap[MAXV],curr[MAXV],prev[MAXV];

int SAP(int src,int des,int n)
{
    int u=src,cnt,sum=0;
    memset(dis,0,sizeof(dis));
    memset(gap,0,sizeof(gap)); gap[0]=n;

    while(dis[src]!=inf)
    {
        bool flag=false;
        for(int i=pre[u];i!=-1;i=next[i])
        {
            if(G[i].w&&dis[u]==dis[G[i].v]+1)
            {
                prev[G[i].v]=u;
                curr[G[i].v]=i;

                flag=true;
                u=G[i].v;

                break;
            }
        }
        if(flag)
        {
            if(u==des)
            {
                cnt=inf;
                for(int i=des;i!=src;i=prev[i])
                {
                    cnt=min(cnt,G[curr[i]].w);
                }
                for(int i=des;i!=src;i=prev[i])
                {
                    G[curr[i]].w-=cnt;
                    G[curr[i]^1].w+=cnt;
                }
                sum+=cnt;
                u=src;
            }
        }
        else
        {
            if(!--gap[dis[u]]) break;
            else
            {
                dis[u]=inf;
                for(int i=pre[u];i!=-1;i=next[i])
                {
                    if(G[i].w) dis[u]=min(dis[u],dis[G[i].v]+1);
                }
                if(dis[u]!=inf) gap[dis[u]]++;
                if(u!=src) u=prev[u];
            }
        }
    }

    return sum;
}

int n,x,y,z,u;
map<int,int> name;

int main()
{
    #ifndef ONLINE_JUDGE
    freopen("Just Another Information Sharing Problem.txt","r",stdin);
    #endif

    while(scanf("%d",&n)!=EOF)
    {
        clear();
        name.clear();

        int cnt=0;
        for(int i=1;n>=i;i++)
        {
            scanf("%d %d %d",&x,&y,&z);
            add(0,i,z);
            while(x--)
            {
                scanf("%d",&u);
                if(!name[u])
                {
                    name[u]=++cnt;
                    add(name[u]+n,MAXV-1,1);
                }
                add(i,name[u]+n,1);
            }
        }
        scanf("%d",&u);
        add(0,u,inf);
        printf("%d\n",SAP(0,MAXV-1,2+n+name.size()));
    }
    return 0;
}