POJ 3255（迪杰斯特拉算法求次短路）

POJ3255，问题是求节点1到n的次短路。
在dijkstra求最短路算法的基础上进行变形，用两个数组分别记录源点到各节点最短路径和次短路径；
每次更新时，都将最短路的节点及可能成为次短路的节点push入队；

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<utility>
using namespace std;
#define maxn 100005
const int INF=0x3f3f3f3f;
typedef pair<int,int> p;

struct edge
{
	int to;
	int next;
	int cost;
}mp[maxn*2];
int pre[maxn],dist[maxn],dist2[maxn];//最短路及次短路; 
int n,m,cnt;
int s,t;

inline void ini()
{
	cnt=-1;
	memset(pre,-1,sizeof(pre));
}

void add(int u,int v,int w)
{
	mp[++cnt].to=v;
	mp[cnt].cost=w;
	mp[cnt].next=pre[u];
	pre[u]=cnt;
}

void dijkstra()
{
	int i,j;
	priority_queue< p, vector<p>, greater<p> >q;
	while(!q.empty())
	{
		q.pop();
	 }
	 memset(dist,INF,sizeof(dist));
	 memset(dist2,INF,sizeof(dist)); 
	dist[s]=0;
	q.push(p(dist[s],s));
	while(!q.empty())
	{
		p k=q.top();
		q.pop();
		//这里不是求最短路,不能到终点就return,还要更新次短路<有可能队列里还有其他元素可以更新次短路; 
		if(dist2[k.second]<k.first)//当前取出的点不是最短路径或者最短路径,continue; 
		{
			continue;
		}
		for(i=pre[k.second];i!=-1;i=mp[i].next)//可能成为次短路的点都进队; 
		{ 
		    int dis=k.first+mp[i].cost;
			if(dis<dist[mp[i].to])//更新最短路,与只求最短路的操作有点不同,这里被最短路更新后的路仍有可能成为次短路,因此要保存起来; 
			{
				swap(dist[mp[i].to],dis);
				q.push(p(dist[mp[i].to],mp[i].to));
			}
			//接下来要更新次短路;
			if(dist[mp[i].to]<dis && dist2[mp[i].to]>dis)//若最短路和次短路中间夹了其他可行路径,则更新次短路; 
			{
				dist2[mp[i].to]=dis;
				q.push(p(dist2[mp[i].to],mp[i].to)); 
			 }  
		}
	}
}

int main()
{
	int i,j;
	int a,b,c;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		ini();
		for(i=1;i<=m;i++)
		{
			scanf("%d%d%d",&a,&b,&c);
			add(a,b,c);
			add(b,a,c);
		}
		s=1;
		t=n;
		dijkstra();
		printf("%d\n",dist2[n]);
	}
	return 0;
}