UVA10806 Dijkstra, Dijkstra. (最小费用最大流)

UVA比较慢，就给个vjudge上的链接吧：
http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=29819
题意：一个无向图，每条边只能走一次，求1到n往返的最短距离，走不到就输出Back to jail。
题目名字很误导思路，一开始就在想最短路一次，然后去掉走过的边，再最短路。但是仔细思考会发现这个其实是不对的，有可能：第一次走的确实是最短路，但是第二次的最短路可能因为第一次把边走掉了变得很远，这种情况还不如折中走两次不是最短路的路径。(个人YY，应该会这样吧)
       思路：每条边流量为1，费用为边权，双向边。建立源点，源点到1的流量为2，费用为0，表示进入2个流量，然后求源点到n的费用流，如果到源点到n的流量不是2的话，说明不通。如果流量是2的话，输出到n点的费用。
代码：

#include<cstdio>
#include<cstring>
#include<string>
#include<queue>
#include<map>
#include<iostream>
#include<vector>
#include<cstdlib>
#include<cmath>
#include<stack>
#include<cctype>
#include<set>
#include<ctime>
#include<cassert>
#include<algorithm>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int INF=1e9+7;
#define MP make_pair
#define PB push_back
const int N=110;
struct EE{
    int from,to,cap,cost;
    EE(){}
    EE(int f,int t,int cap,int cost):from(f),to(t),cap(cap),cost(cost){}
}edge[4*N*N];
int n,m,st,ed,f,ans,Ecnt,d[N],flow[N],pre[N];
bool vis[N];
vector<vector<int> >G;
void init(){
    st=0;ed=n;Ecnt=0;f=0;ans=0;
    G.clear();G.resize(n+1);
}
inline void add(int from,int to,int cap,int cost){
    G[from].PB(Ecnt);
    edge[Ecnt++]=EE(from,to,cap,cost);
    G[to].PB(Ecnt);
    edge[Ecnt++]=EE(to,from,0,-cost);
}
bool spfa(){
    memset(vis,false,sizeof vis);
    fill(d,d+n+1,INF);
    queue<int> Q;Q.push(st);
    vis[st]=true;d[st]=0;pre[st]=-1;flow[st]=INF;
    while(!Q.empty()){
        int s=Q.front();Q.pop();
        vis[s]=false;
        for(int i=0;i<(int)G[s].size();i++){
            EE e=edge[G[s][i]];
            if(e.cap>0&&d[e.to]>d[s]+e.cost){
                d[e.to]=d[s]+e.cost;
                pre[e.to]=G[s][i];
                flow[e.to]=min(flow[s],e.cap);
                if(flow[e.to]==0)break;
                if(!vis[e.to]){vis[e.to]=true;Q.push(e.to);}
            }
        }
    }
    if(d[ed]==INF)return false;
    f+=flow[ed];ans+=flow[ed]*d[ed];
    int now=ed;
    while(now!=st){
        now=pre[now];
        edge[now].cap-=flow[ed];
        edge[now^1].cap+=flow[ed];
        now=edge[now].from;
    }
    return true;
}
int main()
{
//    freopen("D://input.txt","r",stdin);
    while(scanf("%d",&n)!=EOF&&n){
        scanf("%d",&m);
        init();
        add(0,1,2,0);
        for(int i=0;i<m;i++){
            int a,b,c;scanf("%d%d%d",&a,&b,&c);
            add(a,b,1,c);
            add(b,a,1,c);
        }
        while(spfa());
        if(f<2)printf("Back to jail\n");
        else printf("%d\n",ans);
    }
    return 0;
}