BZOJ_4177_Mike的农场_最小割

BZOJ_4177_Mike的农场_最小割

Description

Mike有一个农场，这个农场n个牲畜围栏，现在他想在每个牲畜围栏中养一只动物，每只动物可以是牛或羊，并且每个牲畜围栏中的饲养条件都不同，其中第i个牲畜围栏中的动物长大后，每只牛可以卖a[i]元，每只羊可以卖b[i]元，为了防止牛羊之间相互影响，Mike找到了m条规律，每条规律给出一个三元组(i, j, k)表示如果第i个围栏和第j个围栏养的是不同的动物，那么Mike就需要花费k的代价请人帮忙处理牛羊之间的影响。不过同时Mike也发现k条特殊的规则(S, a, b)，表示如果S中所有牲畜围栏中都养的是动物a，那么Mike可以获得b的额外收入。现在Mike想知道他该在哪些围栏中饲养什么动物才能使得总收益最大，为了简化问题，你只需要输出最大收益。

Input

第一行三个整数n、m、k，表示一共有n个围栏，m条规律，k条规则。

第二行有n个整数，表示a[i]。

第三行有n个整数，表示b[i]。

接下来m行，每行有三个整数(i, j, k)表示一条规则。

再接下来k行，每行一开始有三个整数t、a和b，表示一条规则(S, a, b)，其中S的大小为t，接下来

t个整数表示S中的元素（a为0表示全为牛，a为1表示全为羊）。

Output

输出一个整数ans，表示最大收益。

Sample Input

4 2 1
1 2 3 1
2 3 1 2
1 2 3
1 3 2
2 0 100 1 2

Sample Output

108

HINT

对于100的数据，n <= 5000, m <= 5000, k <= 5000, a = 0 or 1。

---

经典的最小割建模，目的是把收益最大转化为总收益-最小割。

S->i(a[i]) i->T(b[i])

对于规律(i,j,k),连i->j(k),j->i(k)表示限制。

对于规则(S,a,b),新建一个点tot，如果是牛，则S->tot->点集(b)。否则点集->tot->T(b)

代码：

#include <cstdio>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 50050
#define M 4000500
#define S (n+1)
#define T (n+2)
#define inf 1<<30
int head[N],to[M],nxt[M],flow[M],sum,cnt=1,n,m,K;
int dep[N],Q[N],l,r;
inline void add(int u,int v,int f) {
    to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt; flow[cnt]=f;
    to[++cnt]=u; nxt[cnt]=head[v]; head[v]=cnt; flow[cnt]=0;
}
bool bfs() {
    memset(dep,0,sizeof(dep));
    l=r=0; dep[S]=1; Q[r++]=S;
    while(l<r) {
        int x=Q[l++],i;
        for(i=head[x];i;i=nxt[i]) {
            if(!dep[to[i]]&&flow[i]) {
                dep[to[i]]=dep[x]+1;
                if(to[i]==T) return 1;
                Q[r++]=to[i];
            }
        }
    }
    return 0;
}
int dfs(int x,int mf) {
    if(x==T) return mf;
    int nf=0,i;
    for(i=head[x];i;i=nxt[i]) {
        if(dep[to[i]]==dep[x]+1&&flow[i]) {
            int tmp=dfs(to[i],min(mf-nf,flow[i]));
            if(!tmp) dep[to[i]]=0;
            nf+=tmp;
            flow[i]-=tmp;
            flow[i^1]+=tmp;
            if(nf==mf) break;
        }
    }
    return nf;
}
void dinic() {
    int f;
    while(bfs()) {
        while(f=dfs(S,inf)) {
            sum-=f;
        }
    }
    printf("%d\n",sum);
}
int main() {
    scanf("%d%d%d",&n,&m,&K);
    int tot=n+2,i,x,y,z,w;
    for(i=1;i<=n;i++) scanf("%d",&x),sum+=x,add(S,i,x);
    for(i=1;i<=n;i++) scanf("%d",&x),sum+=x,add(i,T,x);
    for(i=1;i<=m;i++) {
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z); add(y,x,z);
    }
    for(i=1;i<=K;i++) {
        scanf("%d%d%d",&x,&y,&z);
        tot++;
        while(x--) {
            scanf("%d",&w);
            if(!y) add(tot,w,inf);
            else add(w,tot,inf);
        }
        if(!y) add(S,tot,z);
        else add(tot,T,z);
        sum+=z;
    }
    dinic();
}