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=1,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,-1,sizeof(lev));
    lev[S]=0;
    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]==-1 && e[i].w>0)
         lev[v]=lev[x]+1,q.push(v);
        }
        q.pop();
    }
    return lev[T]!=-1;
}
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]+1 && e[i].w>0 && (tmp=dfs(v,min(flw,e[i].w)))>0)
        {
            e[i].w-=tmp;
            e[i^1].w+=tmp;
            return tmp;
        }
    }
    return 0;
}
int main()
{
    scanf("%d%d",&m,&n);
    S=n+1,T=n+2;
    for (int i=1;i<=m;i++)
    scanf("%d",&a[i]);
    for (int i=1;i<=n;i++)
    {
    scanf("%d",&k);
    for (int j=1;j<=k;j++)
    {
        scanf("%d",&x);
        buy[i].push_back(x);
    }
    scanf("%d",&x);
    add(i,T,x),add(T,i,0);
    }
    for (int i=1;i<=n;i++)
    for (int j=0;j<buy[i].size();j++)
    {
        int v=buy[i][j];
        if (!last[v])
        last[v]=i,add(S,i,a[v]),add(i,S,0);
        else
        add(last[v],i,INF),add(i,last[v],0),last[v]=i;
    }
    while (bfs())
    while ((tmp=dfs(S,INF))>0) ans+=tmp;
    printf("%d",ans);
    return 0;
}