标签:style blog http color os io for art
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 ; }
标签:style blog http color os io for art
原文地址:http://www.cnblogs.com/xiaohongmao/p/3873957.html