Bzoj1458:士兵占领:带上下界的网络流

题目链接：1458:士兵占领

上手就是一发最小流

每个行和列看成一个点，行连边(S,i)，下界l[i]上界inf，列连边(i+n,T)，下界c[i]上界inf，如果格子(i,j)无障碍连边(i,j+n)，下界0上界1(普通边)

然后最小流模板

然后就痛快的wa了QAQ

调了半天不过，要来数据一测，NND为什么有个崩溃？

好吧数组开小了……

#include<queue>
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=310;
const int inf=0x7fffffff/3;
int n,m,k,h[maxn],tot=1,cur[maxn];
struct edge{int to,next,w;}G[maxn*maxn*2];
int vis[maxn],du[maxn],S,SS,T,TT;
bool flag[maxn][maxn];

void add(int x,int y,int z){
G[++tot].to=y; G[tot].next=h[x]; h[x]=tot; G[tot].w=z;
G[++tot].to=x; G[tot].next=h[y]; h[y]=tot; G[tot].w=0;
}

bool bfs(){
for (int i=1;i<=TT;++i) vis[i]=-1;
queue<int>q; q.push(SS); vis[SS]=0;
while (!q.empty()){
int u=q.front(); q.pop();
for (int i=h[u];i;i=G[i].next){
int v=G[i].to;
if (vis[v]==-1&&G[i].w){
vis[v]=vis[u]+1;
q.push(v);
}
}
}return vis[TT]!=-1;
}

int dfs(int x,int f){
if (x==TT||f==0) return f;
int w,used=0;
for (int i=cur[x];i;i=G[i].next)
if (vis[G[i].to]==vis[x]+1){
w=f-used;
w=dfs(G[i].to,min(w,G[i].w));
G[i].w-=w; G[i^1].w+=w;
used+=w; if (G[i].w) cur[x]=i;
if (used==f) return f;
}
if (!used) vis[x]=-1;
return used;
}

int dinic(){
int ret=0;
while (bfs()){
for (int i=1;i<=TT;++i) cur[i]=h[i];
ret+=dfs(SS,inf);
}return ret;
}

int main(){
scanf("%d%d%d",&n,&m,&k);
S=n+m+1,T=S+1,SS=T+1,TT=SS+1;
for (int i=1;i<=n;++i){
int x; scanf("%d",&x);
du[S]-=x; du[i]+=x;
add(S,i,inf);
}
for (int i=1;i<=m;++i){
int x; scanf("%d",&x);
du[i+n]-=x; du[T]+=x;
add(i+n,T,inf);
}
for (int i=1;i<=k;++i){
int x,y; scanf("%d%d",&x,&y);
flag[x][y]=1;
}
for (int i=1;i<=n;++i)
for (int j=1;j<=m;++j)
if (!flag[i][j]) add(i,j+n,1);
for (int i=1;i<=TT;++i)
if (du[i]>0) add(SS,i,du[i]);
else if (du[i]<0) add(i,TT,-du[i]);
add(T,S,inf); int tmp=0; dinic();
for (int i=h[SS];i;i=G[i].next)
if (G[i].w){
printf("JIONG!");
return 0;
}
for (int i=h[TT];i;i=G[i].next)
if (G[i^1].w){
printf("JIONG!");
return 0;
}
for (int i=h[T];i;i=G[i].next)
if (G[i].to==S){
tmp+=G[i^1].w;G[i].w=G[i^1].w=0;break;}
add(SS,T,inf); add(S,TT,inf);
printf("%d",tmp-dinic());
}