POJ 1149 PIGS | 最大流问题

参考了这个PDF

第一道网络流啊!感动

 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #include<vector>
 #include<queue>
 #define INF 100000000
 #define N 500
 #define M 1100
 using namespace std;
 struct adj
 {
     int u,v,w;
 }e[M*M];
 int head[N],n,m,S,T,a[M],k,ecnt=,x,flw[N],tmp,lev[N],ans,last[M];
 vector <int> buy[M];
 queue <int> q;
 void add(int u,int v,int w)
 {
     e[++ecnt].v=v;
     e[ecnt].w=w;
     e[ecnt].u=head[u];
     head[u]=ecnt;
 }
 int bfs()
 {
     memset(lev,-,sizeof(lev));
     lev[S]=;
     q.push(S);
     while (!q.empty())
     {
         int x=q.front();
         for (int i=head[x];i;i=e[i].u)
         {
              int v=e[i].v;
              if (lev[v]==- && e[i].w>)
          lev[v]=lev[x]+,q.push(v);
         }
         q.pop();
     }
     return lev[T]!=-;
 }
 int dfs(int x,int flw)
 {
     if (x==T) return flw;
     for (int i=head[x];i;i=e[i].u)
     {
     int v=e[i].v,tmp;
         if (lev[v]==lev[x]+ && e[i].w> && (tmp=dfs(v,min(flw,e[i].w)))>)
         {
             e[i].w-=tmp;
             e[i^].w+=tmp;
             return tmp;
         }
     }
     return ;
 }
 int main()
 {
     scanf("%d%d",&m,&n);
     S=n+,T=n+;
     for (int i=;i<=m;i++)
     scanf("%d",&a[i]);
     for (int i=;i<=n;i++)
     {
     scanf("%d",&k);
     for (int j=;j<=k;j++)
     {
         scanf("%d",&x);
         buy[i].push_back(x);
     }
     scanf("%d",&x);
     add(i,T,x),add(T,i,);
     }
     for (int i=;i<=n;i++)
     for (int j=;j<buy[i].size();j++)
     {
         int v=buy[i][j];
         if (!last[v])
         last[v]=i,add(S,i,a[v]),add(i,S,);
         else
         add(last[v],i,INF),add(i,last[v],),last[v]=i;
     }
     while (bfs())
     while ((tmp=dfs(S,INF))>) ans+=tmp;
     printf("%d",ans);
     return ;
 }