POJ 2195 最小费用流

点击打开链接

题意：给个乱七八糟的方阵，H代表家，m代表人，现在所有人都要回到一个家，问所有人走到家的步数和

思路：还是很好想到费用流的，费用为人走到家的步数，求最小，流量即为人的个数，连边的话，每个人都连到家的容量为1，费用为步数的边，建立超级源点与人相连，容量为1，费用为0，家与超级汇点相连，一样容量为1肥育馆为0，跑最小费用流就是结果了，PS：入门题，还是蛮简单的.........

#include <queue>
#include <vector>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=10010;
typedef pair<int,int> P;
struct edge{
    int to,cap,cost,rev;
    edge();
    edge(int a,int b,int c,int d){to=a,cap=b,cost=c,rev=d;};
};
vector<edge>G[maxn];
int h[maxn],dis[maxn],prevv[maxn],preve[maxn];
void addedge(int st,int en,int cap,int cost){
    G[st].push_back(edge(en,cap,cost,G[en].size()));
    G[en].push_back(edge(st,0,-cost,G[st].size()-1));
}
int min_cost_flow(int st,int en,int f){
    int ans=0;
    memset(h,0,sizeof(h));
    while(f>0){
        priority_queue<P,vector<P>,greater<P> >que;
        memset(dis,inf,sizeof(dis));
        dis[st]=0;que.push(P(0,st));
        while(!que.empty()){
            P p=que.top();que.pop();
            int v=p.second;
            if(dis[v]<p.first) continue;
            for(unsigned int i=0;i<G[v].size();i++){
                edge &e=G[v][i];
                if(e.cap>0&&dis[e.to]>dis[v]+e.cost+h[v]-h[e.to]){
                    dis[e.to]=dis[v]+e.cost+h[v]-h[e.to];
                    prevv[e.to]=v;
                    preve[e.to]=i;
                    que.push(P(dis[e.to],e.to));
                }
            }
        }
        if(dis[en]==inf) return -1;
        for(int i=0;i<maxn;i++) h[i]+=dis[i];
        int d=f;
        for(int i=en;i!=st;i=prevv[i]){
            d=min(d,G[prevv[i]][preve[i]].cap);
        }
        f-=d;
        ans+=d*h[en];
        for(int i=en;i!=st;i=prevv[i]){
            edge &e=G[prevv[i]][preve[i]];
            e.cap-=d;
            G[i][e.rev].cap+=d;
        }
    }
    return ans;
}
int num1[110][2],num2[110][2];
int main(){
    int n,m;
    char str[110][110];
    while(scanf("%d%d",&n,&m)!=-1){
        if(n==0&&m==0) break;
        memset(num1,0,sizeof(num1));
        memset(num2,0,sizeof(num2));
        for(int i=0;i<maxn;i++) G[i].clear();
        for(int i=0;i<n;i++) scanf("%s",str[i]);
        int k1=0,k2=0;
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                if(str[i][j]=='m'){
                    num1[k1][0]=i;num1[k1++][1]=j;
                }else if(str[i][j]=='H'){
                    num2[k2][0]=i;num2[k2++][1]=j;
                }
            }
        }
        for(int i=1;i<=k1;i++) addedge(0,i,1,0);
        for(int i=k1+1;i<=k1+k2;i++) addedge(i,k1+k2+1,1,0);
        for(int i=0;i<k1;i++){
            for(int j=0;j<k2;j++){
                int t=abs(num2[j][0]-num1[i][0])+abs(num2[j][1]-num1[i][1]);
                addedge(i+1,j+k1+1,1,t);
            }
        }
        printf("%d\n",min_cost_flow(0,k1+k2+1,k1));
    }
    return 0;
}