POJ-2159 最小費用最大流

                                                    Going Home
node

 本身寫的第一道費用流,圖建好一波板子AC。不過仍是有幾個地方有點迷。spa

 先來看看思想:題意是平面圖上每一個點有若干我的和若干個房子,保證人的數量等於房子的數量。每一個格子上能夠同時容納全部人,也容許一我的在一個房子的格子上但不進入房子。求全部人都找到一個房子住所走的距離的和的最小值。code

  典型的最小費用最大流問題,用0點做爲源點,給人編號,而後從源點引入一條邊鏈接人,容量爲1,費用爲0,給房子編號,每一個人到每一個房子連一條邊,容量爲1,費用爲他們的曼哈頓距離。而後再設置一個點做爲匯點,全部的房子都與匯點連一條邊,容量爲1,費用爲0.求一遍最小費用流就好了。blog

const int N=1e5+10;
struct Edge
{
    int to,next,cap,flow,cost;
} e[N*10];
int head[N],tot;
int pre[N],dis[N];
int vis[N],Vis[201][201];
int n,m,tn;
char s[201][201];
void init(int num)
{
    tot=0;//邊的數量
    tn=num;
    memset(head,-1,sizeof(head));
}
void add(int u,int v,int cap,int cost)
{
    e[tot].to=v,e[tot].cap=cap,e[tot].cost=cost,e[tot].flow=0;
    e[tot].next=head[u];
    head[u]=tot++;
    e[tot].to=u,e[tot].cap=0,e[tot].cost=-cost,e[tot].flow=0;
    e[tot].next=head[v];
    head[v]=tot++;
}
bool spfa(int s,int t)
{
    queue<int>q;
    for(int i=0; i<=tn; i++)
    {
        dis[i]=INF;
        vis[i]=0;
        pre[i]=-1;
    }
    dis[s]=0;
    vis[s]=1;
    q.push(s);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        vis[u]=0;
        for(int i=head[u]; i+1; i=e[i].next)
        {
            int v=e[i].to;
            if(e[i].cap>e[i].flow&&dis[v]>dis[u]+e[i].cost)
            {
                dis[v]=dis[u]+e[i].cost;
                pre[v]=i;
                if(!vis[v])
                {
                    vis[v]=1;
                    q.push(v);
                }
            }
        }
    }
    if(pre[t]==-1)  return false;
    return true;
}
int mincost_maxflow(int s,int t)
{
    int cost=0;
    while(spfa(s,t))
    {
        int Min=INF;
        for(int i=pre[t]; i!=-1; i=pre[e[i^1].to])
            if(Min>e[i].cap-e[i].flow)
                Min=e[i].cap-e[i].flow;
        for(int i=pre[t]; i!=-1; i=pre[e[i^1].to])
        {
            e[i].flow+=Min;
            e[i^1].flow-=Min;
            cost+=e[i].cost*Min;
        }
    }
    return cost;
}
struct node
{
    int x,y;
};
vector<node>M,H;
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        if(n==0&&m==0) return 0;
        M.clear();
        H.clear();
        int num=1,tmp=1;
        for(int i=0; i<n; i++)
        {
            scanf("%s",s[i]);
            for(int j=0; j<m; j++) if(s[i][j]!='.') tmp++;
        }
        init(tmp);
        for(int i=0; i<n; i++)
            for(int j=0; j<m; j++)
                if(s[i][j]!='.')
                {
                    if(s[i][j]=='m')
                    {
                        add(0,num,1,0);//0爲源點;
                        M.push_back(node {i,j});
                    }
                    else
                    {
                        add(num,tmp,1,0);//m+h+1爲匯點
                        H.push_back(node {i,j});
                    }
                    Vis[i][j]=num++;
                }
        for(int i=0; i<M.size(); i++)
        {
            node tmpM=M[i];
            for(int j=0; j<H.size(); j++)
            {
                node tmpH=H[j];
                int diss=abs(tmpM.x-tmpH.x)+abs(tmpM.y-tmpH.y);
                add(Vis[tmpM.x][tmpM.y],Vis[tmpH.x][tmpH.y],1,diss);
            }
        }
        printf("%d\n",mincost_maxflow(0,tmp));
    }
    return 0;
}
相關文章
相關標籤/搜索