poj 2195 最小费用最大流

5 5
HH..m
.....
.....
.....
mm..H

简单题目直接代码 ：
#include<iostream>
#include<algorithm>
#include<memory.h>
#include<queue>
#include<cstdio>
using namespace std;
const int M=102;
const int MAXN=10002;
const int inf=0x3f3f;
char map[M][M];
int pre[MAXN];          // pre[v] = k：在增广路上，到达点v的边的编号为k
int dis[MAXN];          // dis[u] = d：从起点s到点u的路径长为d
int vis[MAXN];         // inq[u]：点u是否在队列中
int path[MAXN];
int head[MAXN];
int dx[5]= {1,-1,0,0};
int dy[5]= {0,0,-1,1};
int n,m,NE,sink,tot,ans;
struct node
{
int u,v,cap,cost,next;
} Edge[M*M*10];
void addEdge(int u,int v,int cap,int cost)
{
Edge[NE].u=u;
Edge[NE].v=v;
Edge[NE].cap=cap;
Edge[NE].cost=cost;
Edge[NE].next=head[u];
head[u]=NE++;
Edge[NE].v=u;
Edge[NE].u=v;
Edge[NE].cap=0;
Edge[NE].cost=-cost;
Edge[NE].next=head[v];
head[v]=NE++;
}
int inmap(int x,int y)
{
if(x>=0&&x<n&&y>=0&&y<m) return 1;
return 0;
}
bool SPFA(int s,int t)                   //  源点为0，汇点为sink。
{
int i;
memset(dis,inf,sizeof(dis));
memset(vis,0,sizeof(vis));
memset(pre,-1,sizeof(pre));
dis[s] = 0;
que[s] = 0;
queue<int>q;
q.push(s);
vis[s] =1;
while(!q.empty())        //  这里最好用队列，有广搜的意思，堆栈像深搜。
{
int u =q.front();
q.pop();
for(i = head[u]; i != -1; i = Edge[i].next)
{
int v = Edge[i].v;
if(Edge[i].cap >0&& dis[v] > dis[u] + Edge[i].cost)
{
dis[v] = dis[u] + Edge[i].cost;
pre[v] = u;
path[v]=i;
if(!vis[v])
{
vis[v] =1;
q.push(v);
}
}
}
vis[u] =0;
}
if(pre[t]==-1)
return false;
return true;
}
void end(int s,int t)
{
int u, p,sum = inf;
for(u=t; u!=0; u=pre[u])
{
sum = min(sum,Edge[path[u]].cap);
}
for(u = t; u != 0; u=pre[u])
{
Edge[path[u]].cap -= sum;
Edge[path[u]^1].cap += sum;
ans += sum*Edge[path[u]].cost;     //  cost记录的为单位流量费用，必须得乘以流量。
}
}
int main()
{
int i,j,k,cas=1;
while(scanf("%d%d",&n,&m),(n||m))
{
NE=0,tot=0,ans=0;
sink=n*m+1;
memset(head,-1,sizeof(head));
for(i=0; i<n; i++)
{
for(j=0; j<m; j++)
{
cin>>map[i][j];
}
}
for(i=0; i<n; i++)
{
for(j=0; j<m; j++)
{
for(k=0; k<4; k++)
{
int x=i+dx[k];
int y=j+dy[k];
if(inmap(x,y))
{
addEdge(j+1+i*m,x*m+y+1,n,1);
}
}
if(map[i][j]=='m')
{
addEdge(0,i*m+j+1,1,0);
}
else if(map[i][j]=='H')
{
addEdge(i*m+j+1,sink,1,0);
}
}
}
while(SPFA(0,sink))
{
end(0,sink);
}
printf("%d\n",ans);
}
return 0;
}
/*
2 2
.m
H.
5 5
HH..m
.....
.....
.....
mm..H
7 8
...H....
...H....
...H...H
mmm.mmmm
...H....
...H....
...H....
7 8
........
........
HHHHHHH.
mmmmmm..
.....m..
........
........

*/