HDU 3488

http://acm.hdu.edu.cn/showproblem.php?pid=3488

原来写过的一道题，今天重新看费用流又做了一遍

题意：给一个图，求环的并（权值和最小）

思路：每个点只能走一次，且都要走，所以一个点的出度入度均为1，因此拆点建图跑二分图最优匹配

用费用流写的，速度比km慢

#include <iostream>

#include <cstdio>

#include <cstring>

#include <queue>

using namespace std ;

const int INF=0xfffffff ;

struct node{

    int s,t,cap,cost,nxt ;

}e[200005] ;

int sumflow ;

int n,m,cnt,head[1005],vis[1005],dis[1005],pre[1005] ;

void add(int s,int t,int cap,int cost)

{

    e[cnt].s=s ;e[cnt].t=t ;e[cnt].cap=cap ;e[cnt].cost=cost ;e[cnt].nxt=head[s] ;head[s]=cnt++ ;

    e[cnt].s=t ;e[cnt].t=s ;e[cnt].cap=0 ;e[cnt].cost=-cost ;e[cnt].nxt=head[t] ;head[t]=cnt++ ;

}

int spfa(int s,int t,int N)

{

    for(int i=0 ;i<=N ;i++)

        dis[i]=INF ;

    dis[s]=0 ;

    memset(vis,0,sizeof(vis)) ;

    memset(pre,-1,sizeof(pre)) ;

    vis[s]=1 ;

    queue <int> q ;

    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].nxt)

        {

            int tt=e[i].t ;

            if(e[i].cap && dis[tt]>dis[u]+e[i].cost)

            {

                dis[tt]=dis[u]+e[i].cost ;

                pre[tt]=i ;

                if(!vis[tt])

                {

                    vis[tt]=1 ;

                    q.push(tt) ;

                }

            }

        }

    }

    if(dis[t]==INF)return 0 ;

    return 1 ;

}

int MCMF(int s,int t,int N)

{

    int flow,minflow,mincost ;

    mincost=flow=0 ;

    while(spfa(s,t,N))                                                

    {

        minflow=INF ;

        for(int i=pre[t] ;i!=-1 ;i=pre[e[i].s])

            minflow=min(minflow,e[i].cap) ;

        flow+=minflow ;

        for(int i=pre[t] ;i!=-1 ;i=pre[e[i].s])

        {

            e[i].cap-=minflow ;

            e[i^1].cap+=minflow ;

        }

        mincost+=dis[t]*minflow ;

    }

    sumflow=flow ;//最大流 

    return mincost ;

}

int main()

{

    int t ;

    scanf("%d",&t) ;

    while(t--)

    {

        cnt=0 ;

        memset(head,-1,sizeof(head)) ;

        scanf("%d%d",&n,&m) ;

        int ss=0 ;

        int st=2*n+1 ;

        for(int i=0 ;i<m ;i++)

        {

            int s,t,v ;

            scanf("%d%d%d",&s,&t,&v) ;

            add(s,t+n,1,v) ;

        }

        for(int i=1 ;i<=n ;i++)

            add(0,i,1,0) ;

        for(int i=n+1 ;i<=2*n ;i++)

            add(i,st,1,0) ;

        int ans=MCMF(ss,st,st+1) ;

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

    }

    return 0 ;

}

View Code