题面
描述
狼爱上羊啊爱的疯狂,谁让他们真爱了一场;
狼爱上羊啊并不荒唐,他们说有爱就有方向……
Orez 听到这首歌,心想:狼和羊如此和谐,为什么不尝试羊狼合养呢?说干就干! Orez 的羊狼圈可以看作一个 n*m 个矩阵格子,这个矩阵的边缘已经装上了篱笆。可是 Drake 很快发现狼再怎么也是狼,它们总是对羊垂涎三尺,那首歌只不过是一个动人的传说而已。所以 Orez 决定在羊狼圈中再加入一些篱笆,还是要将羊狼分开来养。 通过仔细观察,Orez 发现狼和羊都有属于自己领地,若狼和羊们不能呆在自己的领地,那它们就会变得非常暴躁,不利于他们的成长。 Orez 想要添加篱笆的尽可能的短。当然这个篱笆首先得保证不能改变狼羊的所属领地,再就是篱笆必须修筑完整,也就是说必须修建在单位格子的边界上并且不能只修建一部分。
输入格式
文件的第一行包含两个整数 n 和 m。接下来 n 行每行 m 个整数,1 表示该格子属于狼的领地,2 表示属于羊的领地,0 表示该格子不是任何一只动物的领地。
输出格式
文件中仅包含一个整数 ans,代表篱笆的最短长度。
思路
考虑建网格图($(x,y)$ 到 $(x-1,y)$,$(x+1,y)$,$(x,y-1)$,$(x,y+1)$ 建边),要让狼到羊之间无法到达,就是在求一个最小割。
$S$ 到每只狼(或羊)连边,每只狼(或羊)到 $T$ 连边,权值为 $+ \infty$,跑一遍最大流(最大流 = 最小割)。
代码
#include<bits/stdc++.h>
using namespace std;
long long n,m,k,last[200005],to[200005],nextt[200005],s[200005],top=1;
long long v[200005],d[200005];
void add(int a,int b,int c){
nextt[++top]=last[a];
to[top]=b;
s[top]=c;
last[a]=top;
}
bool bfs(long long S,long long T){
memset(v,0,sizeof(v));
memset(d,63,sizeof(d));
queue<long long> q;
q.push(S);
v[S]=1;
d[S]=0;
while (!q.empty()){
long long now=q.front();
q.pop();
v[now]=0;
for (int i=last[now];i;i=nextt[i]){
int j=to[i];
if (d[j]>d[now]&&s[i]){
d[j]=d[now]+1;
if (!v[j]){
q.push(j);
v[j]=1;
}
}
}
}
if (d[T]<1000000000){
return 1;
}
return 0;
}
int dfs(long long x,long long minn,long long T){
if (x==T){
return minn;
}
int now=minn,tmp=0;
for (int i=last[x];i&&now;i=nextt[i]){
int j=to[i];
if (d[j]==d[x]+1&&s[i]){
tmp=dfs(j,min(minn,s[i]),T);
if (tmp){
s[i]-=tmp;
s[i^1]+=tmp;
now-=tmp;
}
}
}
return minn-now;
}
long long dinic(long long S,long long T){
long long ans=0,tmp=0;
while (bfs(S,T)){
while (1){
tmp=dfs(S,2147483647,T);
if (!tmp){
break;
}
ans+=tmp;
}
}
return ans;
}
inline int calc(int x,int y){
return (x-1)*m+y;
}
int main(){
long long S=40005,T=40006;
int a[105][105];
cin>>n>>m;
for (int i=1;i<=n;i++){
for (int j=1;j<=m;j++){
cin>>a[i][j];
}
}
for (int i=1;i<=n;i++){
for (int j=1;j<=m;j++){
if (a[i][j]==1){
add(S,calc(i,j),19260817);
add(calc(i,j),S,0);
}
if (a[i][j]==2){
add(calc(i,j),T,19260817);
add(T,calc(i,j),0);
}
if (i<n){
add(calc(i,j),calc(i+1,j),1);
add(calc(i+1,j),calc(i,j),0);
}
if (i>1){
add(calc(i,j),calc(i-1,j),1);
add(calc(i-1,j),calc(i,j),0);
}
if (j<m){
add(calc(i,j),calc(i,j+1),1);
add(calc(i,j+1),calc(i,j),0);
}
if (j>1){
add(calc(i,j),calc(i,j-1),1);
add(calc(i,j-1),calc(i,j),0);
}
}
}
bfs(S,T);
cout<<dinic(S,T);
}