Going Home POJ

来源：互联网发布：网络漫画家朴泰俊编辑：程序博客网时间：2024/06/07 17:53

传送门：POJ2195

题意：给定n*m矩阵，矩阵中的'H'代表房子，'m'代表人，人可以上下左右移动，每移动一次要花一单位费用，每个房子只能住一个人，问每人到达一个房子的最少花费总和是多少。

思路：感觉和以前做过的某个搜索题很像，不过这里同一点可以同时容纳多人，只是一个人房子只能住一个人，将每个房子和每个人建边，流量限制为1，花费为曼哈顿距离，然后就可以用最小费用最大流求解了。

代码：

#include<iostream>#include<queue>#include<string.h>#include<stdio.h>#include<algorithm>#define ll long long#define inf 0x3f3f3f3fusing namespace std;typedef pair<int,int>P;const int MAXN=210;const int MAXM=100010;int head[MAXN],pre[MAXN],dis[MAXN],book[MAXN];int cnt=0,N;//N为点数 struct node{int to,next,cap,flow,cost;}edge[MAXM];void init(){cnt=0;memset(head,-1,sizeof(head));}void addedge(int u,int v,int cap,int cost){edge[cnt].to=v;edge[cnt].cap=cap;edge[cnt].flow=0;edge[cnt].cost=cost;edge[cnt].next=head[u];head[u]=cnt++;edge[cnt].to=u;edge[cnt].cap=0;edge[cnt].flow=0;edge[cnt].cost=-cost;edge[cnt].next=head[v];head[v]=cnt++;}bool spfa(int s,int t){queue<int>q;while(!q.empty())q.pop();for(int i=0;i<=N;i++){dis[i]=inf;pre[i]=-1;book[i]=0;}dis[s]=0;book[s]=1;q.push(s);while(!q.empty()){int u=q.front();q.pop();book[u]=0;for(int i=head[u];i!=-1;i=edge[i].next){int v=edge[i].to;if(edge[i].cap>edge[i].flow&&dis[v]>dis[u]+edge[i].cost){dis[v]=dis[u]+edge[i].cost;pre[v]=i;if(!book[v]){book[v]=1;q.push(v);}} } }return pre[t]!=-1;}int min_cost_max_flow(int s,int t,int &cost){int flow=0;cost=0;while(spfa(s,t)){int temp=inf;for(int i=pre[t];i!=-1;i=pre[edge[i^1].to])temp=min(temp,edge[i].cap-edge[i].flow);for(int i=pre[t];i!=-1;i=pre[edge[i^1].to]){edge[i].flow+=temp;edge[i^1].flow-=temp;cost+=edge[i].cost*temp;}flow+=temp;}return flow;}char mp[110][110];vector<P>H,man;int main(){int n,m;while(cin>>n>>m){if(n+m==0)break;init(); H.clear();man.clear();for(int i=0;i<n;i++){scanf("%s",mp[i]);for(int j=0;j<m;j++){if(mp[i][j]=='H')H.push_back(P(i,j));else if(mp[i][j]=='m')man.push_back(P(i,j));}}int sz=H.size(),szz=man.size();for(int i=0;i<sz;i++){for(int j=0;j<szz;j++){addedge(i+1,j+sz+1,1,abs(H[i].first-man[j].first)+abs(H[i].second-man[j].second));}}N=sz+szz+1;for(int i=1;i<=sz;i++)addedge(0,i,1,0);for(int i=1;i<=szz;i++)addedge(sz+i,N,1,0);int cost=0;min_cost_max_flow(0,N,cost);cout<<cost<<endl;}}

阅读全文

0 0