POJ2195:Going Home(费用流入门）

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <queue>
#include <math.h>
#define inf 0x3f3f3f3f
using namespace std;
int n,m,s,t,tt;
struct pointe
{
int x,y;
} l[1001],r[1001];
struct node
{
int x,y,c,w;
int next;
} eg[20001];
char map[102][102];
int head[1001],v[1001],dis[1001],pre[1001];
void init()
{
memset(head,-1,sizeof(head));
tt=0;
s=0;
t=n*m+1;
}
void add(int xx,int yy,int cc,int ww)
{
eg[tt].x=xx;
eg[tt].y=yy;
eg[tt].c=cc;
eg[tt].w=ww;
eg[tt].next=head[xx];
head[xx]=tt++;
eg[tt].x=yy;
eg[tt].y=xx;
eg[tt].c=0;
eg[tt].w=-ww;
eg[tt].next=head[yy];
head[yy]=tt++;
}
int spfa(int s,int t)
{
for(int i=s; i<=t; i++)
{
v[i]=0;
pre[i]=-1;
dis[i]=inf;
}
queue<int>q;
dis[s]=0;
q.push(s);
while(!q.empty())
{
int ff=q.front();
q.pop();
v[ff]=0;
//今天算是刚了解邻接边表第一次更新的是与源点相邻的第一层
//例如head[10]->1,3,5,7....;
for(int i=head[ff]; i!=-1; i=eg[i].next)
{
int w1=eg[i].y;
if(eg[i].c&&dis[w1]>dis[ff]+eg[i].w)
{
dis[w1]=dis[ff]+eg[i].w;//注意这里dis[ff]
pre[w1]=i;//算是了解么意思了
if(!v[w1])
{
v[w1]=1;
q.push(w1);
}
}
}
}
if(dis[t]==inf)
{
return 0;
}
return 1;
}
void KM(int s,int t)
{
int ans=0,minx;
while(spfa(s,t)==1)
{
minx=inf;
for(int i=pre[t]; i!=-1; i=pre[eg[i].x])
minx=min(minx,eg[i].c);
for(int i=pre[t]; i!=-1; i=pre[eg[i].x])
{
eg[i].c-=minx;
eg[i+1].c+=minx;
}
ans+=minx*dis[t];
}
cout<<ans<<endl;
return ;
}
int main()
{
int t1,t2,temp;
while(scanf("%d%d",&n,&m)!=EOF&&(n||m))
{
t1=t2=0;
init();
for(int i=1; i<=n; i++)
scanf("%*c%s",map[i]+1);
for(int i=1; i<=n; i++)
{
for(int j=1; j<=m; j++)
{
temp=(i-1)*m+j;
if(map[i][j]=='m')
{
add(s,temp,1,0);
l[t1].x=i;
l[t1++].y=j;
}
else if(map[i][j]=='H')
{
add(temp,t,1,0);
r[t2].x=i;
r[t2++].y=j;
}
}
}
for(int i=0; i<t1; i++)
{
int l1=(l[i].x-1)*m+l[i].y;
for(int j=0; j<t2; j++)
{
int l2=(r[j].x-1)*m+r[j].y;
int l3=abs(l[i].x-r[j].x)+abs(l[i].y-r[j].y);
add(l1,l2,1,l3);
}
}
KM(s,t);
}
return 0;
}

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <queue>
#include <math.h>
#define inf 0x3f3f3f3f
using namespace std;
struct node1//记录h或者m的坐标
{
int x,y;
} a[402],b[402];
struct node
{
int x,y,c,w;
int next;
} eg[600001];
int n,m,tt,head[10001],v[10001],dis[10001],pre[10001],s,t;
char map[102][102];
void init()
{
tt=0;
memset(head,-1,sizeof(head));
s=0;
t=n*m+1;
}
void add(int xx,int yy,int cc,int ww)
{
eg[tt].x=xx;
eg[tt].y=yy;
eg[tt].c=cc;
eg[tt].w=ww;
eg[tt].next=head[xx];
head[xx]=tt++;
eg[tt].x=yy;
eg[tt].y=xx;
eg[tt].c=0;
eg[tt].w=-ww;
eg[tt].next=head[yy];
head[yy]=tt++;
}
int spfa(int s,int t)
{
queue<int>q;
for(int i=0; i<=t; i++)
{
v[i]=0;
dis[i]=inf;
pre[i]=-1;
}
v[s]=1;
dis[s]=0;
q.push(s);
while(!q.empty())
{
int ff=q.front();
q.pop();
v[ff]=0;
for(int i=head[ff]; i!=-1; i=eg[i].next)
{
int w1=eg[i].y;
if(eg[i].c&&dis[w1]>dis[ff]+eg[i].w)
{
dis[w1]=dis[ff]+eg[i].w;
pre[w1]=i;//用边表建图存储其信息
if(!v[w1])
{
v[w1]=1;
q.push(w1);
}
}
}
}
if(dis[t]==inf)
{
return 0;
}
return 1;
}
void KM(int s,int t)
{
int min1,ans=0;
while(spfa(s,t)==1)
{
min1=inf;
for(int i=pre[t]; i!=-1; i=pre[eg[i].x])
{
if(min1>=eg[i].c)
{
min1=eg[i].c;
}
}
for(int i=pre[t]; i!=-1; i=pre[eg[i].x])
{
eg[i].c-=min1;
eg[i+1].c+=min1;
}
ans+=min1*dis[t];
}
printf("%d\n",ans);
return ;
}
int main()
{
int t1,t2;
while(scanf("%d%d",&n,&m)!=EOF)
{
if(n==0&&m==0) break;
init();
t1=0;
t2=0;
for(int i=1; i<=n; i++)
{
scanf("%s",map[i]+1);//因为要创建超级源点，所以从一比较好计算
for(int j=1; j<=m; j++)
{
int l=(i-1)*m+j;//以点建图
if(map[i][j]=='m')
{
add(s,l,1,0);
a[t1].x=i;
a[t1++].y=j;
}
else if(map[i][j]=='H')
{
add(l,t,1,0);
b[t2].x=i;
b[t2++].y=j;
}
}
}
for(int i=0; i<t1; i++)
{
int l1=(a[i].x-1)*m+a[i].y;//以点建图
for(int j=0; j<t2; j++)
{
int l2=abs(a[i].x-b[j].x)+abs(a[i].y-b[j].y);
int l3=(b[j].x-1)*m+b[j].y;//以点建图
add(l1,l3,1,l2);
}
}
KM(s,t);
}
return 0;
}