[2019CCPC秦皇岛] E Escape 最大流

给出一个 n × m n×m n×m的矩阵, n , m ≤ 100 n,m\leq100 n,m100。给出 a a a个入口,表示有 a a a个机器人从这些入口进来。给出 b b b个出口,表示机器人可以从这 b b b个出口出去。一些格子是障碍点表示无法通行,然后你可以在每个格子最多设置一个转向机关让机器人转向。转向分别可以转化左下,右下,左上,右上。求问是否最终 a a a个机器人都能走出去。
因为无论是转向或者是不转向。如果两个机器人的路径重复了,那么最终只能同时从一个出口出去,所以机器人的路径不能重复。
每个点拆成两个点分别代表水平和竖直方向合,在两点之间连流量为1的双向边表示转向。源点连所有入口的竖直点流量为1,出口点连汇点流量为1。判断是否满流。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int N=1e2+3;
char s[N][N];
int mp[N][N],tot=0;
namespace Dinic {
	const int N=3e4,M=3e5; 
	int front[N],head[N],dep[N],tot=-1,n,S,T;
	struct edge{ int v,nxt;ll w; }e[2*M];
	void init() {
		tot=-1;
		for(int i=1;i<=n;i++) front[i]=-1;
	}
	void add(int u,int v,ll w) {
		e[++tot].v=v,e[tot].w=w,e[tot].nxt=front[u],front[u]=tot;
		e[++tot].v=u,e[tot].w=0,e[tot].nxt=front[v],front[v]=tot;
	}
	bool bfs() {
		for(int i=1;i<=n;i++) dep[i]=0;
		queue<int> q;
		q.push(S);
		dep[S]=1;
		while(!q.empty()) {
			int u=q.front();
			q.pop();
			for(int i=front[u];i!=-1;i=e[i].nxt) {
				int v=e[i].v;
				if(dep[v]==0&&e[i].w) {
					dep[v]=dep[u]+1;
					q.push(v); 
				}
			}
		}
		return dep[T];
	}
	ll dfs(int cur,ll dis=1e18) {
		if(cur==T) return dis;
		ll flow=0,sum=0;
		for(int &i=head[cur];i!=-1;i=e[i].nxt) {
			int v=e[i].v;
			if(dep[v]==dep[cur]+1&&e[i].w) {
				flow=dfs(v,min(dis,e[i].w));
				dis-=flow,sum+=flow;
				e[i].w-=flow,e[i^1].w+=flow;
				if(!dis) break;
			}
		}
		return sum;
	}
	ll solve() {
		ll ans=0;
		while(bfs()) {
			for(int i=1;i<=n;i++) head[i]=front[i];
			ans+=dfs(S);
		}
		return ans;
	}
}
bool check(int x,int y) {
	if(s[x][y]=='1') return 0;
	else return 1;
}
int in[N],out[N]; 
int main() {
	int T;
	scanf("%d",&T);
	while(T--) {
		int n,m,a,b;
		scanf("%d%d%d%d",&n,&m,&a,&b);
		Dinic::n=2*n*m+2;
		int S=Dinic::S=2*n*m+1;
		int T=Dinic::T=2*n*m+2;
		Dinic::init(); 
		for(int i=1;i<=n;i++)  
			scanf("%s",s[i]+1);
		for(int i=1;i<=a;i++) scanf("%d",&in[i]);
		for(int i=1;i<=b;i++) scanf("%d",&out[i]);
		tot=0;
		for(int i=1;i<=n;i++) {
			for(int j=1;j<=m;j++) {
				mp[i][j]=++tot;
			}
		}
		for(int i=1;i<=n;i++) {
			for(int j=1;j<=m;j++) {
				if(s[i][j]=='1') continue; 
				if(j-1>=1&&check(i,j-1)) Dinic::add(mp[i][j],mp[i][j-1],1); 
				if(j+1<=m&&check(i,j+1)) Dinic::add(mp[i][j],mp[i][j+1],1);
				if(i-1>=1&&check(i-1,j)) Dinic::add(mp[i][j]+tot,mp[i-1][j]+tot,1);
				if(i+1<=n&&check(i+1,j)) Dinic::add(mp[i][j]+tot,mp[i+1][j]+tot,1);
				Dinic::add(mp[i][j],mp[i][j]+tot,1);
				Dinic::add(mp[i][j]+tot,mp[i][j],1);
			}
		}
		for(int i=1;i<=a;i++)  {
			Dinic::add(S,mp[1][in[i]]+tot,1);
		}
		for(int i=1;i<=b;i++) {
			Dinic::add(mp[n][out[i]]+tot,T,1);
		}
		ll ans=Dinic::solve();
		if(ans==a) puts("Yes");
		else puts("No");  
	}
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值