[P2774] 方格取数问题(最大流)

将所有点以横纵坐标和为奇数或偶数分成两部分,建立二分图,把相邻的点都连边,左右两边分别与源点和汇点相连。因为不能取相邻的点,所以要去掉一些边使剩余的图不含s到t的路径。又要留下的点权和最大,所以需要去掉的点权和最小,所以是求最小割。因为最小割=最大流,所以让每个点和源点或汇点的边容量为点权,点和点之间的边容量为inf,求最大流,再用所有点权和减最大流即可。

#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=10010,M=80000,inf=0x7fffffff;
struct edge{
	int y,f,next;
}data[M];
int n,m,s,t,sum,num,h[N],dep[N],cur[N];
queue<int> q;
inline int read(){
	int x=0,f=0;
	char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return f?-x:x;
} 
inline void add(int x,int y,int f){
	data[++num].y=y;data[num].f=f;data[num].next=h[x];h[x]=num;
	data[++num].y=x;data[num].f=0;data[num].next=h[y];h[y]=num;
}
inline bool bfs(int s,int t){
	memset(dep,0,sizeof dep);
	for(int i=s;i<=t;i++)cur[i]=h[i];
	while(!q.empty())q.pop();
	q.push(s);dep[s]=1;
	while(!q.empty()){
		int u=q.front();q.pop();
		for(int i=h[u];i!=-1;i=data[i].next){
			int v=data[i].y;
			if(!dep[v]&&data[i].f>0){
				dep[v]=dep[u]+1;
				q.push(v);
			}
		}
	}
	return dep[t];
}
int dfs(int u,int t,int lim){
	if(u==t)return lim;
	int res=0;
	for(int& i=cur[u];i!=-1;i=data[i].next){
		int v=data[i].y;
		if(dep[v]==dep[u]+1&&data[i].f>0){
			int fmax=dfs(v,t,min(lim-res,data[i].f));
			if(fmax>0){
				data[i].f-=fmax;data[i^1].f+=fmax;
				res+=fmax;
				if(res==lim)return res;
			}
		}
	}
	return res;
}
inline int dinic(int s,int t){
	int flow=0;
	while(bfs(s,t))flow+=dfs(s,t,inf);
	return flow;
}
int main(){
	n=read();m=read();s=0;t=n*m+1;
	memset(h,-1,sizeof h);num=-1;
	for(int i=1;i<=n;i++)
	for(int x,w,j=1;j<=m;j++){
		x=(i-1)*m+j;w=read();sum+=w;
		if((i+j)%2){
			add(s,x,w);
			if(i-1>0)add(x,x-m,inf);
			if(j-1>0)add(x,x-1,inf);
			if(i+1<=n)add(x,x+m,inf);
			if(j+1<=m)add(x,x+1,inf);
		}
		else add(x,t,w);
	}
	printf("%d",sum-dinic(s,t));
	return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
下面是Python实现网络流方格取数问题代码示例: ```python from ortools.graph import pywrapgraph # 定义方格的大小和初始值 grid_size = 4 grid = [[4, 7, 8, 6], [6, 7, 3, 9], [3, 5, 2, 1], [9, 2, 8, 2]] # 创建网络流图 max_flow = pywrapgraph.SimpleMaxFlow() source = 0 sink = grid_size ** 2 + 1 node_num = grid_size ** 2 + 2 # 添加源点和汇点 max_flow.AddNode(node_num) max_flow.AddArc(source, sink, float('inf')) # 添加方格中的节点和边 for i in range(grid_size): for j in range(grid_size): node_id = i * grid_size + j + 1 max_flow.AddNode(node_num) max_flow.AddArc(source, node_id, grid[i][j]) if i == 0: max_flow.AddArc(node_id, sink, 0) else: for k in range(grid_size): next_node_id = (i - 1) * grid_size + k + 1 max_flow.AddArc(node_id, next_node_id, float('inf')) if j == 0: max_flow.AddArc(node_id, sink, 0) else: for k in range(grid_size): next_node_id = i * grid_size + k + 1 max_flow.AddArc(node_id, next_node_id, float('inf')) # 求解最大流 status = max_flow.Solve(source, sink) if status == max_flow.OPTIMAL: max_flow_value = max_flow.OptimalFlow() print('最大取数和为:', max_flow_value) ``` 该代码使用了Google OR-Tools库中的`SimpleMaxFlow`模块,实现了方格取数问题最大流解法。该算法将方格中的每个元素看作一个节点,并且将每个节点与其上、下、左、右的相邻节点之间的边的容量设置为无限大,确保每个节点都可以通过其相邻节点到达汇点。同时,将源点与每个节点之间的边的容量设置为该节点的初始值,保证源点只能向该节点中取出其初始值大小的数。最终通过求解网络流问题,得到了最大取数和。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值