POJ 2195 Going Home(KM或费用流)

题意:给一个字符矩阵,m表示man,H表示house。人和房子个数相等。一个房子只能住一个人,要求所有人走到空的房子,每走一步花费1,要求总的花费最小,输出最小值。

解法1:KM,给定一个带权的二分图,求权值最大的完美匹配。O(N^3),N为人的数量

题意要求总花费最小,用KM只需要把花费弄成相反数(负数),再运行KM算法,求出后结果是一个负的最大值,输出它的相反数就是正的最小了。

#include <iostream>
#include <stdio.h>
#include <math.h>
#include <algorithm>
#include <queue>
#include <stack>
#include <vector>
#include <string>
#include <string.h>
#include <map>
#include <set>
using namespace std;
#define maxn 505
#define inff 1<<30
char ch[maxn][maxn];
struct node
{
    int x,y;
}mat[maxn];
int e[maxn][maxn];
int n,m,tot;
int abs(int x)
{
    if(x<0)
    return -x;
    else return x;
}
int t[maxn],s[maxn],lx[maxn],ly[maxn],r[maxn];
int match(int i)
{
    int j;
    s[i]=1;
    for(j=1;j<tot;j++)
    {
        if(lx[i]+ly[j]==e[i][j]&&!t[j])
        {
            t[j]=1;
            if(!r[j]||match(r[j]))
            {
                r[j]=i;
                return 1;
            }
        }
    }
    return 0;
}
void update()
{
    int a=inff;
    for(int i=1;i<tot;i++)
    {
        if(s[i])
        for(int j=1;j<tot;j++)
            if(!t[j])a=min(a,lx[i]+ly[j]-e[i][j]);
    }
    for(int i=1;i<tot;i++)
    {
        if(s[i])
            lx[i]-=a;
        if(t[i])
            ly[i]+=a;
    }
}
void km()
{
    int i,j;
   for(i=1;i<tot;i++)
   {
       r[i]=lx[i]=ly[i]=0;
       for(j=1;j<tot;j++)
       {
           lx[i]=max(lx[i],e[i][j]);
       }
   }
   for(i=1;i<tot;i++)
   {
       while(1)
       {
           memset(s,0,sizeof s);
           memset(t,0,sizeof t);
           if(match(i))break;
           else update();

       }
   }
}
int main()
{
    int i,j,num;
   while(scanf("%d%d",&n,&m)!=EOF)
   {
       if(n==0&&m==0)break;
       tot=1;
       memset(e,0,sizeof e);
       for(i=0;i<n;i++)
       {
           scanf("%s",ch[i]);
           for(j=0;j<m;j++)
            if(ch[i][j]=='m')mat[tot].x=i,mat[tot++].y=j;
       }
       num=tot;
       for(i=0;i<n;i++)
        for(j=0;j<m;j++)
        if(ch[i][j]=='H')
        mat[num].x=i,mat[num++].y=j;
       for(i=1;i<tot;i++)
       {
           for(j=tot;j<num;j++)
           {
               e[i][j-tot+1]=-abs(mat[i].x-mat[j].x)-abs(mat[i].y-mat[j].y);
               //cout<<e[i][j-tot+1]<<" ";
           }
          // cout<<endl;
       }

       km();
       int ans=0;
       for(j=1;j<tot;j++)
       {
            ans+=e[r[j]][j];
       }

       printf("%d\n",-ans);

   }
  return 0;
}

解法2:费用流,建立一个超级源点(0)和一个总汇点(num+1,num是最大点的编号)。

源点和人的容量为1,花费为 0

房子和汇点的容量为1,花费为0

人和房子花费(距离)其实就是他们横纵坐标对应相减再相加,容量为1。

因为所有容量为1,时间复杂度为O(min(n^2/3,m^1/2)*m),n为总的点数,m为总的边数

#include<iostream>
#include<string>
#include<stdio.h>
#include<string.h>
#include<vector>
#include<math.h>
#include<queue>
#include<map>
#include<set>
#include<algorithm>
using namespace std;
#define MAXN 505
#define MAXM 505*500*2
#define LL long long
#define INF 0x3f7f7f7f
const double eps = 1e-10;
struct Edge
{
    int from,to,cap,flow,cost;
    Edge(int from=0,int to=0,int cap=0,int flow=0,int cost=0)
    {
        this->from=from;
        this->to=to;
        this->cap=cap;
        this->flow=flow;
        this->cost=cost;
    }
};
vector<Edge>edges;
vector<int>g[MAXN];
bool vis[MAXN];
int dis[MAXN];
int pre[MAXN];
int a[MAXN];
int n,m,s,t;
void init()
{
    for(int i=0;i<MAXN;i++)
    {
        g[i].clear();
    }
    edges.clear();
}
void add(int from,int to,int cap,int cost)
{
    edges.push_back(Edge(from,to,cap,0,cost));
    edges.push_back(Edge(to,from,0,0,-cost));
    int m=edges.size();
    g[from].push_back(m-2);
    g[to].push_back(m-1);
}
bool spfa(int &flow,int &cost)
{
    int i;
    for(i=0;i<MAXN;i++)
        dis[i]=INF;
    memset(vis,0,sizeof(vis));
    queue<int>q;
    q.push(s);
    dis[s]=0;
    vis[s]=1;
    pre[s]=0;
    a[s]=INF;
   while(!q.empty())
   {
       int u=q.front();
       q.pop();
       vis[u]=0;
       int sz=g[u].size();
       for(i=0;i<sz;i++)
       {
           Edge &e=edges[g[u][i]];
           if(e.cap>e.flow&&dis[e.to]>dis[u]+e.cost)
           {
               dis[e.to]=dis[u]+e.cost;
               pre[e.to]=g[u][i];
               a[e.to]=min(a[u],e.cap-e.flow);
               if(!vis[e.to])
               {
                   vis[e.to]=1;
                   q.push(e.to);
               }
           }
       }
   }
   if(dis[t]==INF)
    return false;
   flow+=a[t];
   cost+=dis[t]*a[t];
   int v=t;
   while(v!=s)
   {
       edges[pre[v]].flow+=a[t];
       edges[pre[v]^1].flow-=a[t];
       v=edges[pre[v]].from;
   }
   return true;
}

int minCost()
{
    int flow=0,cost=0;
    while(spfa(flow,cost));
    return cost;
}
struct node
{
    node(int x=0,int y=0):x(x),y(y){}
    int x,y;
};
int ABS(int x)
{
    return x>0?x:-x;
}
struct node p[MAXN*MAXN];
char maze[MAXN][MAXN];
int main()
{
    int i,j,tot,num;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        if(n==0&&m==0)break;
        tot=num=0;
        for(i=1;i<=n;i++)
        {
            scanf("%s",maze[i]+1);
            for(j=1;j<=m;j++)
            {
                if(maze[i][j]=='m')
                    p[++tot]=node(i,j);
            }
        }
        num=tot;
        for(i=1;i<=n;i++)
        {
            for(j=1;j<=m;j++)
            {
                if(maze[i][j]=='H')
                    p[++num]=node(i,j);
            }
        }
        init();
        for(i=1;i<=tot;i++)
        {
            for(j=tot+1;j<=num;j++)
            {
                int len=ABS(p[i].x-p[j].x)+ABS(p[i].y-p[j].y);
               // cout<<"\\"<<len<<endl;
                add(i,j,1,len);
            }
        }
        for(i=1;i<=tot;i++)
            add(0,i,1,0);
        for(i=tot+1;i<=num;i++)
            add(i,num+1,1,0);
        s=0,t=num+1;
        cout<<minCost()<<endl;
    }
    return 0;
}

 


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值