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
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(); }