题意:从1到n再到1,每条边只能走一次,求最短距离。
建图:每条边只能走一次就是流量是1,添加源点与1相连,容量为2,费用为0,n与汇点相连容量为2,费用为0;
求增广路用SPFA最短路求,,
#include<stdio.h> #include<queue> #include<string.h> const int N=1100; const int inf=0x3fffffff; using namespace std; int cost[N],start,end,n,head[N],num,pre[N],vis[N]; struct edge { int st,ed,cp,flow,next; }e[N*N]; void addedge(int x,int y,int c,int w) { e[num].st=x;e[num].ed=y;e[num].cp=c; e[num].flow=w;e[num].next=head[x];head[x]=num++; e[num].st=y;e[num].ed=x;e[num].cp=-c;e[num].flow=0;e[num].next=head[y];head[y]=num++; } int SPFA() { int i,u,v; queue<int>Q; for(i=start;i<=end;i++) {cost[i]=inf;vis[i]=0;pre[i]=-1;} cost[start]=0;vis[start]=1; Q.push(start); while(!Q.empty()) { u=Q.front(); Q.pop();vis[u]=0; for(i=head[u];i!=-1;i=e[i].next) { v=e[i].ed; if(e[i].flow>0&&cost[v]>cost[u]+e[i].cp) { pre[v]=i; cost[v]=cost[u]+e[i].cp; if(vis[v]==0) { vis[v]=1; Q.push(v); } } } } if(pre[end]==-1) return 0; return 1; } int mincost() { int MINcost=0,maxflow=0,i,minflow; while(SPFA()) { minflow=inf; for(i=pre[end];i!=-1;i=pre[e[i].st]) if(minflow>e[i].flow) minflow=e[i].flow; maxflow+=minflow; for(i=pre[end];i!=-1;i=pre[e[i].st]) { e[i].flow-=minflow; e[i^1].flow+=minflow; MINcost+=e[i].cp; } } return MINcost; } int main() { int i,x,y,c,m; while(scanf("%d%d",&n,&m)!=-1) { start=0;end=n+1;num=0; memset(head,-1,sizeof(head)); for(i=0;i<m;i++) { scanf("%d%d%d",&x,&y,&c); addedge(x,y,c,1); addedge(y,x,c,1); } addedge(start,1,0,2); addedge(n,end,0,2); printf("%d ",mincost()); } return 0; }