题意:给定n个谷仓,有m条双向路径,每条连接两个谷仓。一个人要从1号谷仓走到n号谷仓,再从n号谷仓走回1号(来回走的路不能重复)。给定路径长度,问来回走的最短长度。
思路:由于来回可以看成2条从1到n的不同的路,所以转化为求从1到n的两条不同的路。按照最小费用流建图:假设a b之间有长度为c的路,那么ab之间费用为c,容量是1;ba之间费用为c,容量是1。(同时要建立b到a的容量为0,费用为-c的路径;也就是输入给出一条边,需要添加到邻接表中四条边)接着建立一个超级源点,连接1号景点,无费用,容量为2;同理建立一个超级汇点,连接n号景点,无费用,容量为2。
针对这道题,更新路径值的时候实际上可以断定流量只能增加1(因为每条边的流量最多为1),而且题目保证有解,进而可知spfa必执行两次,所以更新函数可以简写(见版本2)
#include <stdio.h> #include <string.h> #define N 1005 #define M 10005 #define min(a,b) a<b?a:b #define INF 0x3fffffff struct edge{ int x,y,cost,cap,next; }e[M*4]; int n,m,top,res; int dis[N],visited[N],pre[N],q[200000],index[N],first[N]; void add(int x,int y,int c,int w){ e[top].y = y; e[top].cap = c; e[top].cost = w; e[top].next = first[x]; first[x] = top++; e[top].y = x; e[top].cap = 0; e[top].cost = -w; e[top].next = first[y]; first[y] = top++; } int spfa(){//费用当做权值,求最短路 int i,front,rear,now; for(i = 0;i<=n;i++) dis[i] = INF; memset(visited,0,sizeof(visited)); memset(pre,-1,sizeof(pre)); front = rear = -1; q[++rear] = 0; dis[0] = 0; while(front < rear){ now = q[++front]; visited[now] = 0; for(i = first[now];i!=-1;i=e[i].next) if(e[i].cap && dis[e[i].y]>dis[now]+e[i].cost){ dis[e[i].y] = dis[now]+e[i].cost; if(!visited[e[i].y]){ visited[e[i].y] = 1; q[++rear] = e[i].y; } index[e[i].y] = i;//保存松弛的节点在邻接表中的位置 pre[e[i].y] = now; } } if(dis[n]!=INF) return 1; return 0; } void sum(){ int i,aug=INF; for(i = n;i!=0;i=pre[i])//得到此次增加的容量 aug = min(aug,e[index[i]].cap); res += dis[n]*aug; for(i = n;i!=0;i=pre[i]){ e[index[i]].cap -= aug; e[index[i]^1].cap += aug; } } int main(){ freopen("a.txt","r",stdin); while(scanf("%d %d",&n,&m)!=EOF){ int i,a,b,c; top = res = 0; memset(first,-1,sizeof(first)); for(i = 0;i<m;i++){ scanf("%d %d %d",&a,&b,&c); add(a,b,1,c); add(b,a,1,c); } add(0,1,2,0);//添加超级源点 add(n,n+1,2,0);//添加超级汇点 n++; while(spfa())//每求一次最短路,更新一下图的容量 sum(); printf("%d\n",res); } return 0; }
版本2:
#include <cstdio> #include <cstring> #include <algorithm> #include <cmath> #include <queue> #include <cstdlib> using namespace std; #define INF 0x3fffffff #define N 1005 #define M 40005 struct edge{ int y,next,w,c; }e[M]; int n,m,first[N],top,dis[N],used[N],pre[N],id[N],res; void add(int x,int y,int w,int c){ e[top].y = y; e[top].w = w; e[top].c = c; e[top].next = first[x]; first[x] = top++; } int spfa(){ int i,now; for(i = 1;i<=n+1;i++) dis[i] = INF; dis[0] = 0; queue<int> q; q.push(0); memset(used, 0, sizeof(used)); used[0] = 1; while(!q.empty()){ now = q.front(); q.pop(); used[now] = 0; for(i = first[now];i!=-1;i=e[i].next){ if(e[i].c>0 && dis[e[i].y] > dis[now]+e[i].w){ dis[e[i].y] = dis[now]+e[i].w; if(!used[e[i].y]){ used[e[i].y] = 1; q.push(e[i].y); } id[e[i].y] = i; pre[e[i].y] = now; } } } return dis[n+1]<INF; } void update(){ int i; res+=dis[n+1]; for(i = n+1;i!=0;i=pre[i]){ e[id[i]].c --; e[id[i]^1].c ++; } } int main(){ while(scanf("%d %d",&n,&m)!=EOF){ int i,a,b,w; res = top = 0; memset(first, -1, sizeof(first)); for(i = 1;i<=m;i++){ scanf("%d %d %d",&a,&b,&w); add(a,b,w,1); add(b,a,-w,0); add(b,a,w,1); add(a,b,-w,0); } add(0,1,0,2); add(1,0,0,0); add(n,n+1,0,2); add(n+1,n,0,0); while(spfa()) update(); printf("%d\n",res); } return 0; }