最小费用最大流2

#include <stdio.h>
#include <iostream>
#include <string.h>
#include<cmath>
using namespace std;
const int N=300;
const int MAXE=200000;
const int inf=1<<30;
int head[N],ep;
int d[N],pre[N];
bool vis[N];
int q[MAXE];
struct Edge
{
    int u,v,c,w,next;
}edge[MAXE];
void addedge(int u,int v,int w,int c)//u v 费用 容量
{
    edge[ep].u=u;
    edge[ep].v=v;
    edge[ep].w=w;
    edge[ep].c=c;
    edge[ep].next=head[u];
    head[u]=ep++;
    edge[ep].v=u;
    edge[ep].u=v;
    edge[ep].w=-w;
    edge[ep].c=0;
    edge[ep].next=head[v];
    head[v]=ep++;
}
int SPFA(int src,int des)
{
    int l,r;
    memset(pre,-1,sizeof(pre));
    memset(vis,0,sizeof(vis));
    for(int i=0;i<=des;i++) d[i]=inf;
    d[src]=0;
    l=0;r=0;
    q[r++]=src;
    vis[src]=1;
    while(l<r)
    {
        int u=q[l++];
        vis[u]=0;
        for(int j=head[u];j!=-1;j=edge[j].next)
        {
            int v=edge[j].v;
            if(edge[j].c>0&&d[u]+edge[j].w<d[v])
            {
                d[v]=d[u]+edge[j].w;
                pre[v]=j;
                if(!vis[v])
                {
                    vis[v]=1;
                    q[r++]=v;
                }
            }
        }
    }
    if(d[des]==inf)
        return 0;
    return 1;
}
int MCMF(int src,int des)
{
    int flow=0,ans=0;//flow是得到的最大流的值 ans得到的是最小的费用
    while(SPFA(src,des))
    {
        ans+=d[des];
        int u=des;
        int mini=inf;
        while(u!=src)
        {
            if(edge[pre[u]].c<mini)
                mini=edge[pre[u]].c;
                u=edge[pre[u]].u;
        }
        flow+=mini;
        u=des;
        while(u!=src)
        {
            edge[pre[u]].c-=mini;
            edge[pre[u]^1].c+=mini;
            u=edge[pre[u]].u;
        }
    }
    return ans;
}
struct man
{
    int x;
    int y;
}M[111];
struct house
{
    int x,y;
}H[111];
char str[111];
int main()
{
    int n,m,mcnt,hcnt,i,j,src,des;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        if(!n&&!m) break;
        ep=0;
        memset(head,-1,sizeof(head));
        mcnt=hcnt=0;
        for(i=0;i<n;i++)
        {
            scanf("%s",str);
            for(j=0;j<m;j++)
            {
                if(str[j]=='H')
                {
                    hcnt++;
                    H[hcnt].x=i; H[hcnt].y=j;
                }
                else if(str[j]=='m')
                {
                    mcnt++;
                    M[mcnt].x=i; M[mcnt].y=j;
                }
            }
        }
        for(i=1;i<=mcnt;i++)
        {
            for(j=1;j<=hcnt;j++)
            {
                int dis=abs(M[i].x-H[j].x)+abs(M[i].y-H[j].y);
                addedge(i,mcnt+j,dis,1);
                //addedge(mcnt+j,i,dis,1);
            }
        }
        src=0;
        des=mcnt+hcnt+1;
        for(i=1;i<=mcnt;i++)
          addedge(src,i,0,1);
        for(j=1;j<=hcnt;j++)
            addedge(mcnt+j,des,0,1);
        int ans=MCMF(src,des);

        printf("%d\n",ans);
    }
    return 0;
}

 

#include <stdio.h>
#include <iostream>
#include <string.h>
#include<math.h>

#define N 300
#define MAXE 200000
#define inf 1<<30
int head[N],ep;            //ep记录的是加入的边数
int d[N],pre[N];
bool vis[N];
int q[MAXE];
struct Edge
{
    int u,v,c,w,next;
}edge[MAXE];
void addedge(int u,int v,int w,int c)     // 加边操作,u,v分别是起点和终点,w 记录 路径长度,c记录权值,此题中权值为1
{
    edge[ep].u=u;
    edge[ep].v=v;
    edge[ep].w=w;
    edge[ep].c=c;
    edge[ep].next=head[u];
    head[u]=ep++;
    edge[ep].v=u;
    edge[ep].u=v;
    edge[ep].w=-w;
    edge[ep].c=0;
    edge[ep].next=head[v];
    head[v]=ep++;
}
int SPFA(int src,int des)                         // SPFA 最短路算法,维护一个队列,scr源点,des汇点
{
    int l,r;
    memset(pre,-1,sizeof(pre));                   //  pre[] 数组记录前驱节点
    memset(vis,0,sizeof(vis));
    for(int i=0;i<=des;i++) d[i]=inf;             //  初始化数据
    d[src]=0;                                     //  将源点出队列,并能够多次进入队列
    l=0;r=0;
    q[r++]=src;
    vis[src]=1;
    while(l<r)
    {
        int u=q[l++];
        vis[u]=0;
        for(int j=head[u];j!=-1;j=edge[j].next)
        {
            int v=edge[j].v;
            if(edge[j].c>0&&d[u]+edge[j].w<d[v])         //   判断条件和spfa不同,加入了edge[j].c>0,保证了联通性
            {
                d[v]=d[u]+edge[j].w;
                pre[v]=j;
                if(!vis[v])
                {
                    vis[v]=1;
                    q[r++]=v;
                }
            }
        }
    }
    if(d[des]==inf)
        return 0;
    return 1;
}
int MCMF(int src,int des)
{
    int flow=0,ans=0;               //flow是得到的最大流的值 ans得到的是最小的费用
    while(SPFA(src,des))            // SPFA 判断是否还有增广路径
    {
        ans+=d[des];                // 让ans + 上该次增广路径求得的最大能增广的所需的最小费用
        int u=des;
        int mini=inf;
        while(u!=src)
        {
            if(edge[pre[u]].c<mini)
                mini=edge[pre[u]].c;
                u=edge[pre[u]].u;
        }
        flow+=mini;                    // 加上本次增广路径得到的,最小的流量(短板效应)
        u=des;                         //利用回溯,改变流量
        while(u!=src)
        {
            edge[pre[u]].c-=mini;
            edge[pre[u]^1].c+=mini;
            u=edge[pre[u]].u;
        }
    }
    return ans;                          //返回费用
}
struct man                              // 定义一个M[]结构体,记录人的位置信息
{
    int x;
    int y;
}M[111];
struct house
{
    int x,y;
}H[111];                                // 定义一个H[]结构体,记录人的位置信息
char str[111];
int main()
{
    int n,m,mcnt,hcnt,i,j,src,des;        // hcnt,记录房子的个数,mcnt记录人的数目
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        if(!n&&!m) break;
        ep=0;
        memset(head,-1,sizeof(head));     //初始化ep,head【】,mcnt,hcnt;
        mcnt=hcnt=0;
        for(i=0;i<n;i++)
        {
            scanf("%s",str);              //输入信息
            for(j=0;j<m;j++)
            {
                if(str[j]=='H')
                {
                    hcnt++;
                    H[hcnt].x=i; H[hcnt].y=j;         // H[],数组记录房子的坐标
                }
                else if(str[j]=='m')
                {
                    mcnt++;
                    M[mcnt].x=i; M[mcnt].y=j;        // M[],数组记录人的坐标
                }
            }
        }
        for(i=1;i<=mcnt;i++)
        {
            for(j=1;j<=hcnt;j++)
            {
                int dis=fabs(M[i].x-H[j].x)+fabs(M[i].y-H[j].y);   // dis求得 第i个人和第j个人房屋的距离。
                addedge(i,mcnt+j,dis,1);            //加入i,mcnt+j,这条边,dis 记录路径长度和加入权值1;
                //addedge(mcnt+j,i,dis,1);
            }
        }
        src=0;
        des=mcnt+hcnt+1;
        for(i=1;i<=mcnt;i++)
          addedge(src,i,0,1);             //表明源点和第i个人的距离为0,初始化每一个人和源点的距离;
        for(j=1;j<=hcnt;j++)
            addedge(mcnt+j,des,0,1);      //表明汇点和第j座房屋的距离为0,初始化每一房屋和汇点的距离;
        int ans=MCMF(src,des);            // 求得ans记录从源点src到终点des的最小费用最大流

        printf("%d\n",ans);
    }
    return 0;
}

 

posted @ 2014-08-20 14:43  HuberyQian  阅读(168)  评论(0编辑  收藏  举报