[bzoj3698]XWW的难题

传送门

有源汇上下界最大流,建模很经典,将每行和每列看做一个点,然后去按照有源汇上下界建图就行了,因为这个题,我又发现了自己网络流板子里的一个错误写法,bfs的时候得增广完再退出,改了我半个小时。。。

#include<cstdio>
#include<queue>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define rg register
void read(int &x) {
	char ch; bool ok;
	for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;
	for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x;
}
queue<int>q;int cnt=1,n,s,t,ans,inf=2e9,ss,sum,tt,dis[10001],in[10001],nxt[20001],pre[20001],h[10001],v[20001],cur[10001];double a[101][101];
inline void add(int x,int y,int z)
{
	pre[++cnt]=y,nxt[cnt]=h[x],h[x]=cnt,v[cnt]=z;
	pre[++cnt]=x,nxt[cnt]=h[y],h[y]=cnt,v[cnt]=0;
}
inline bool bfs(int s,int t)
{
	memset(dis,0,sizeof dis);
	q.push(s),dis[s]=1;
	while(!q.empty())
	{
		int x=q.front();q.pop();
		for(rg int i=h[x];i;i=nxt[i])if(v[i]&&!dis[pre[i]])dis[pre[i]]=dis[x]+1,q.push(pre[i]);
	}
	return dis[t];
}
inline int dfs(int x,int flow,int t)
{
	if(x==t||!flow)return flow;
	int f=flow;
	for(rg int &i=cur[x];i;i=nxt[i])
		if(v[i]&&dis[pre[i]]==dis[x]+1)
		{
			int y=dfs(pre[i],min(v[i],f),t);
			f-=y,v[i]-=y,v[i^1]+=y;
			if(!f)return flow;
		}
	if(flow==f)dis[x]=-1;
	return flow-f;
}
int main()
{
	scanf("%d",&n),s=0,t=2*n+1,ss=t+1,tt=ss+1;
	for(rg int i=1;i<=n;i++)for(rg int j=1;j<=n;j++)scanf("%lf",&a[i][j]);
	for(rg int i=1;i<n;i++)
		for(rg int j=1;j<n;j++)
		{
			if(a[i][j]!=(int)a[i][j])add(i,j+n,1);
			in[i]-=(int)a[i][j],in[j+n]+=(int)a[i][j];
		}
	for(rg int i=1;i<n;i++)
	{
		if(a[i][n]!=(int)a[i][n])add(s,i,1);
		in[i]+=(int)a[i][n],in[s]-=(int)a[i][n];
	}
	for(rg int i=1;i<n;i++)
	{
		if(a[n][i]!=(int)a[n][i])add(i+n,t,1);
		in[t]+=(int)a[n][i],in[i+n]-=(int)a[n][i];
	}
	for(rg int i=0;i<=t;i++)
	{
		if(in[i]>0)add(ss,i,in[i]),sum+=in[i];
		if(in[i]<0)add(i,tt,-in[i]);
	}
	add(t,s,inf);
	for(;bfs(ss,tt);ans+=dfs(ss,inf,tt))memcpy(cur,h,sizeof cur);
	if(sum!=ans){printf("No\n");return 0;}
    for(int i=h[ss];i;i=nxt[i])v[i]=v[i^1]=0;
    for(int i=h[tt];i;i=nxt[i])v[i]=v[i^1]=0;
	ans=0;
	for(;bfs(s,t);ans+=dfs(s,inf,t))memcpy(cur,h,sizeof cur);
	printf("%d\n",3*ans);
}
posted @ 2019-01-05 14:07  蒟蒻--lichenxi  阅读(445)  评论(0编辑  收藏  举报