（最小割）洛谷P4313文理分科

会飞的小蛇

于 2020-11-19 17:41:38 发布

阅读量153

点赞数

分类专栏：图论

本文链接：https://blog.csdn.net/Z7784562/article/details/109819069

版权

图论专栏收录该内容

72 篇文章 0 订阅

订阅专栏

洛谷P4313文理分科

思路：

一下子就可以想到最小割，先不考虑一起选的情况的话，就是文理二者择其一的选择。
$S$ 向 $(i, j)$ 建边，权值为 $art_{i,j}$ ；
$(i, j)$ 向 $T$ 建边，权值为 $science_{i,j}$ 。
考虑共同选理科，那么我们可以新建一个点 $i,j)^{'}$ , $S$ 向 $i,j)^{'}$ 连边，权值为 $same\_art_{i,j}$ ,然后 $i,j)^{'}$ 向自身和相邻点连边，权值为 $i n f$ 。这样可以保证在所有需要一起选课的 $(i, j)$ 和 $T$ 相连边全部被割掉的时候，也就是全部选择理科的时候才能够保证这条边不被割。
对于共同选择文科的处理方式相同，建新点 $i,j)^{''}$ ,与 $T$ 连边，权值为 $same\_science_{i,j}$ ，自身和相邻点与 $i,j)^{''}$ 连边，权值为 $i n f$ 。
最后答案就是总和减去最小割。

代码：

#include<bits/stdc++.h>
#define pii pair<int,int>
#define int long long
#define cl(x,y) memset(x,y,sizeof(x))
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
const int N=3e6+210;
const int M=5e2+10;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
const int minn=0xc0c0c0c0;
using namespace std;
struct edge
{
	int u,v,w;
}maze[N<<1];
int len=1,head[N]={0};
int dep[N];//深度 
void add(int u,int v,int w)
{
	maze[++len]={head[u],v,w};
	head[u]=len;
}
void inc(int u,int v,int w)
{
	add(u,v,w);
	add(v,u,0);
} 
int dfs(int u,int f,int t)
{
	int ans=0,i;
	if(u==t)
		return f;
	for(i=head[u];i && f;i=maze[i].u)
	{
		int v=maze[i].v,w=maze[i].w;
		if(dep[v]==dep[u]+1 && w)//符合深度关系且能流 
		{
			int sum=dfs(v,min(f,w),t);
			maze[i].w-=sum;
			maze[i^1].w+=sum;
			f-=sum;
			ans+=sum;
		}	
	}
	if(!ans)
		dep[u]=-2;
	return ans;
}
int bfs(int s,int t)
{
	queue<int> q;
	cl(dep,0);
	dep[s]=1;//源点深度为1
	q.push(s);
	while(!q.empty())
	{
		int u=q.front(),i;
		q.pop();
		for(i=head[u];i;i=maze[i].u)
		{
			int v=maze[i].v,w=maze[i].w;
			if(w && !dep[v])//有深度且能流 
			{
				dep[v]=dep[u]+1;
				q.push(v); 
			}
		}
	}
	return dep[t];
}
int dinic(int s,int t)
{
	int ans=0;
	while(bfs(s,t))
		ans+=dfs(s,inf,t);
	return ans;
}
int a[M][M],b[M][M],c[M][M],d[M][M];
int dis[5][2]={{1,0},{-1,0},{0,1},{0,-1},{0,0}};
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	int n,m,i,j,k;
	cin>>n>>m;
	int sum=0;
	for(i=1;i<=n;i++)
		for(j=1;j<=m;j++)
			cin>>a[i][j];
	for(i=1;i<=n;i++)
		for(j=1;j<=m;j++)
			cin>>b[i][j];
	for(i=1;i<=n;i++)
		for(j=1;j<=m;j++)
			cin>>c[i][j];
	for(i=1;i<=n;i++)
		for(j=1;j<=m;j++)
			cin>>d[i][j];
	for(i=1;i<=n;i++)
		for(j=1;j<=m;j++)
			sum+=a[i][j]+b[i][j]+c[i][j]+d[i][j];
	int s=0,t=3*n*m+1;
	for(i=1;i<=n;i++)
		for(j=1;j<=m;j++)
		{
			inc(s,(i-1)*m+j,a[i][j]);
			inc((i-1)*m+j,t,b[i][j]);
			inc(s,(i-1)*m+j+n*m,c[i][j]);
			inc((i-1)*m+j+2*n*m,t,d[i][j]);
			for(k=0;k<5;k++)
			{
				int dx=i+dis[k][0],dy=j+dis[k][1];
				if(dx>=1 && dx<=n && dy>=1 && dy<=m)
				{
					inc((i-1)*m+j+n*m,(dx-1)*m+dy,inf);
					inc((dx-1)*m+dy,(i-1)*m+j+2*n*m,inf);
				}
			}
		}
	cout<<sum-dinic(s,t)<<endl;
	return 0;
}