POJ-3255-Roadblocks

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
using namespace std;
const int inf=1<<29;
const int maxn=5e3+100;
const int maxm=2e5+100;
int n,m,e,head[maxn],nxt[maxm],cost[maxm],pnt[maxm],dist[maxn][2];
bool vis[maxn];
queue<int> q;
void AddEdge(int u,int v,int c)
{
pnt[e]=v;nxt[e]=head[u];cost[e]=c;head[u]=e++;
pnt[e]=u;nxt[e]=head[v];cost[e]=c;head[v]=e++;
}
void Spfa(int st,int des,int pos)
{
for(int i=0;i<=n;i++)
dist[i][pos]=inf;
dist[st][pos]=0;
q.push(st);
while(!q.empty())
{
int u=q.front();
vis[u]=0;
q.pop();
for(int i=head[u];i!=-1;i=nxt[i])
if(dist[pnt[i]][pos]>dist[u][pos]+cost[i])
{
dist[pnt[i]][pos]=dist[u][pos]+cost[i];
if(!vis[pnt[i]])
{
q.push(pnt[i]);
vis[pnt[i]]=1;
}

}
}
}
void solve()
{
int ans=inf;
Spfa(1,n,0);
Spfa(n,1,1);
for(int i=0;i<e;i++)
if(dist[pnt[i^1]][0]+dist[pnt[i]][1]+cost[i]!=dist
[0])
{
ans=min(ans,dist[pnt[i^1]][0]+cost[i]+dist[pnt[i]][1]);
}
printf("%d\n",ans);

}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
e=0;
memset(head,-1,sizeof(head));
for(int i=0;i<m;i++)
{
int u,v,c;
scanf("%d%d%d",&u,&v,&c);
AddEdge(u,v,c);
}
solve();
}
return 0;
}