BZOJ4177: Mike的农场

BZOJ4177

$MD$考的时候还想半天，这$TM$不就是文理分科吗。真的$sb$了。赶紧退役算了。。
$S$向$n$个点连边$（S,i,a_i）$
$n$个点向$T$连边$（i,T,b_i）$
对于$m$每条规则$(i,j,k)$，连边$(i,j,k)$和$(j,i,k)$
$k$个额外收益，如果是都取$a$获得收益，就建新点，$S$连向新点，然后该点向集合内每个点连边。
如果是$b$，就建新点连向$T$，集合内每个点向该点连边。
$ans=正权和-最小割$

【代码】

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <queue>
#define N 10005
#define M 10000005
#define INF 0x7fffffff
using namespace std;
typedef long long ll;
typedef pair<ll,int> pa;

ll read()
{
    ll x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int n,m,K,cnt=1,T,ans,tot;
int b[M],p[N],nextedge[M],w[M],cur[N];
int Level[N];

void Add(int x,int y,int z) {
    cnt++;
    b[cnt]=y;
    nextedge[cnt]=p[x];
    p[x]=cnt;
    w[cnt]=z;
}

void Anode(int x,int y,int z) {
    Add(x,y,z);Add(y,x,0);
}

bool Bfs()
{
    queue<int>q;
    q.push(0);
    for(int i=1;i<=T;i++) Level[i]=0;
    Level[0]=1;
    while(!q.empty())
    {
        int k=q.front();q.pop();
        for(int i=p[k];i;i=nextedge[i])
        {
            int v=b[i],f=w[i];
            if(!Level[v]&&f)
            {
                Level[v]=Level[k]+1;
                q.push(v);
            }
        }
    }
    return Level[T];
}

int Dfs(int x,int maxf)
{
    if(x==T||!maxf) return maxf;
    int rtn=0;
    for(int i=cur[x];i&&maxf>rtn;i=nextedge[i])
    {
        int v=b[i],f=w[i];
        if(Level[v]==Level[x]+1&&f)
        {
            f=Dfs(v,min(maxf-rtn,f));
            w[i]-=f;w[i^1]+=f;
            rtn+=f;if(w[i]>0) cur[x]=i;
        }
    }
    if(!rtn) Level[x]=0;
    return rtn;
}

int Dinic()
{
    int rtn=0;
    while(Bfs())
    {
        for(int i=0;i<=T;i++) cur[i]=p[i];
        rtn+=Dfs(0,INF);
    }
    return rtn;
}

int main()
{
    n=read(),m=read(),K=read();tot=n;T=n+K+1;
    for(int i=1;i<=n;i++)
    {
        int x=read();
        Anode(0,i,x);ans+=x;
    }
    for(int i=1;i<=n;i++)
    {
        int x=read();
        Anode(i,T,x);ans+=x;
    }
    for(int i=1;i<=m;i++)
    {
        static int x,y,z;
        x=read(),y=read(),z=read();
        Anode(x,y,z);Anode(y,x,z);
    }
    for(int i=1;i<=K;i++)
    {
        static int t,x,y;
        t=read(),x=read(),y=read();ans+=y;
        if(x==0) {
            Anode(0,++tot,y);
            for(int j=1;j<=t;j++){
                int xx=read();Anode(tot,xx,INF);
            }
        }
        else {
            Anode(++tot,T,y);
            for(int j=1;j<=t;j++) {
                int xx=read();Anode(xx,tot,INF);
            }
        }
    }
    printf("%d\n",ans-Dinic());
    return 0;
}