网络流第一题!!!BZOJ1001

时间:2023-03-09 08:27:25
网络流第一题!!!BZOJ1001

歇逼了一晚上,懵懵懂懂的懂了Dinic算法

大概是一遍BFS+DFS,还不是很懂,明天继续看!!!

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn = ;
const int inf = 0x3f3f3f3f;
struct edge{
int to,next,w;
}e[maxn<<];
int n,m,cnt=;
int head[maxn];
int ans;
int d[maxn];
void link(int u,int v,int w){//建立双向边
e[++cnt]=(edge){v,head[u],w};head[u]=cnt;
e[++cnt]=(edge){u,head[v],w};head[v]=cnt;
}
bool bfs(){
memset(d,-,sizeof(d));
queue<int> q;q.push();d[]=;
while(!q.empty()){
int x=q.front();
q.pop();
for (int i=head[x];i;i=e[i].next){
if (e[i].w && d[e[i].to]<){
q.push(e[i].to);
d[e[i].to]=d[x]+;
}
}
}
return d[n*m]< ? :;
}
int dfs(int x,int f){
if (x==n*m || f==)return f;
int w,used=;
for (int i=head[x];i;i=e[i].next)if (e[i].w && d[e[i].to]==d[x]+) {
w=dfs(e[i].to,min(f-used,e[i].w));
e[i].w-=w;
e[i^].w+=w;
used+=w;
if (used==f)return f;
}
if (!used)d[x]=-;
return used;
}
void Dinic(int s){
while(bfs())ans+=dfs(s,inf);
}
int main(){
int x;
scanf("%d%d",&n,&m);
for (int i=;i<=n;i++){
for (int j=;j<m;j++){
scanf("%d",&x);
link(m*(i-)+j,m*(i-)+j+,x);//求出边权连接的左右两个点
}
}
for (int i=;i<n;i++){
for (int j=;j<=m;j++){
scanf("%d",&x);
link(m*(i-)+j,m*i+j,x);//求出边权连接的上下两个点
}
}
for (int i=;i<n;i++){
for (int j=;j<m;j++){
scanf("%d",&x);
link(m*(i-)+j,m*i+j+,x);//斜边上两个点
}
}
Dinic();
printf("%d",ans);
return ;
}