hdu 4411(最小费用流)
题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4411
思路:这道题建图比较难想,首先是建立超级源点和超级汇点,那么由于有K个警察,于是vs与0连边,容量为k,费用为0,因为这K个警察不一定都出去,也就是不一定是最大流,于是0和vt连边,容量为k,费用为0。然后就是拆点建图了:
(1):0到i连边,容量为1,费用0到i的最短路,表示去抓城市i的小偷。
(2):从i到i+n,容量为1,费用为-1000000(很小的负权),这样可以保证每个城市都能遍历到。
(3):对于城市i,j,(i<j),连边(i+n,j),容量为1,费用为i到j的最短路,表示抓完城市i的小偷再去抓城市j的小偷。
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<algorithm> 5 #include<queue> 6 using namespace std; 7 #define MAXN 222 8 #define MAXM 4444444 9 #define inf 1<<30 10 11 struct Edge{ 12 int v,cap,cost,next; 13 }edge[MAXM]; 14 15 int vs,vt,NE,n,m,k; 16 int head[MAXN]; 17 18 void Insert(int u,int v,int cap,int cost) 19 { 20 edge[NE].v=v; 21 edge[NE].cap=cap; 22 edge[NE].cost=cost; 23 edge[NE].next=head[u]; 24 head[u]=NE++; 25 26 edge[NE].v=u; 27 edge[NE].cap=0; 28 edge[NE].cost=-cost; 29 edge[NE].next=head[v]; 30 head[v]=NE++; 31 } 32 33 int dist[MAXN],pre[MAXN],cur[MAXN]; 34 bool mark[MAXN]; 35 bool spfa(int vs,int vt) 36 { 37 memset(mark,false,sizeof(mark)); 38 fill(dist,dist+MAXN-1,inf); 39 dist[vs]=0; 40 queue<int>que; 41 que.push(vs); 42 while(!que.empty()){ 43 int u=que.front(); 44 que.pop(); 45 mark[u]=false; 46 for(int i=head[u];i!=-1;i=edge[i].next){ 47 int v=edge[i].v,cost=edge[i].cost; 48 if(edge[i].cap>0&&dist[u]+cost<dist[v]){ 49 dist[v]=dist[u]+cost; 50 pre[v]=u; 51 cur[v]=i; 52 if(!mark[v]){ 53 mark[v]=true; 54 que.push(v); 55 } 56 } 57 } 58 } 59 return dist[vt]<inf; 60 } 61 62 int MinCostFlow(int vs,int vt) 63 { 64 int flow=0,cost=0; 65 while(spfa(vs,vt)){ 66 int aug=inf; 67 for(int u=vt;u!=vs;u=pre[u]){ 68 aug=min(aug,edge[cur[u]].cap); 69 } 70 flow+=aug,cost+=dist[vt]*aug; 71 for(int u=vt;u!=vs;u=pre[u]){ 72 edge[cur[u]].cap-=aug; 73 edge[cur[u]^1].cap+=aug; 74 } 75 } 76 return cost; 77 } 78 79 int map[MAXN][MAXN]; 80 void floyd() 81 { 82 for(int k=0;k<=n;k++) 83 for(int i=0;i<=n;i++) 84 for(int j=0;j<=n;j++) 85 if(map[i][k]<inf&&map[k][j]<inf&&map[i][k]+map[k][j]<map[i][j]) 86 map[i][j]=map[i][k]+map[k][j]; 87 } 88 89 int main() 90 { 91 int u,v,w; 92 while(~scanf("%d%d%d",&n,&m,&k)){ 93 if(n==0&&m==0&&k==0)break; 94 NE=0; 95 vs=2*n+1,vt=2*n+2; 96 memset(head,-1,sizeof(head)); 97 for(int i=0;i<=n;i++) 98 for(int j=0;j<=n;j++) 99 map[i][j]=(i==j)?0:inf; 100 while(m--){ 101 scanf("%d%d%d",&u,&v,&w); 102 map[u][v]=map[v][u]=min(map[u][v],w); 103 } 104 floyd(); 105 Insert(vs,0,k,0); 106 Insert(0,vt,k,0); 107 for(int i=1;i<=n;i++){ 108 Insert(0,i,1,map[0][i]);//0到各点 109 Insert(i,i+n,1,-1000000);//每个城市拆点 110 Insert(i+n,vt,1,map[0][i]);//各点到汇点,表示回到0 111 } 112 for(int i=1;i<=n;i++){ 113 for(int j=i+1;j<=n;j++){ 114 Insert(i+n,j,1,map[i][j]); 115 } 116 } 117 printf("%d\n",MinCostFlow(vs,vt)+1000000*n); 118 } 119 return 0; 120 }