POJ 3463 Sightseeing(dijkstra)

http://poj.org/problem?id=3463

这道题可以加深对dijkstra的理解；

题意要求找出最短路以及比最短路长1的路径数量；

我们都知道dijkstra是利用每次找出最短的一条边去松弛到别的点距离。

这道题我们就需要每次从起点到别的点的最短路径以及次短路径中找到一个最短的边去松弛起点到别的点的最短路径或者次短路径；

不过就是数组多了一行次短路的状态。

最后判一下次短路的长度是不是最短路+1；    

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stack>
#include<cmath>
using namespace std;
struct node{
	int v,cost,nex;
};
node eg[55555];
int hea[1111];
int dis[1111][2];
int vis[1111][2];
int ans[1111][2];
int n,m;
int e;
void insert(int st,int ed,int cost)
{
	eg[e].v=ed;
	eg[e].cost=cost;
	eg[e].nex=hea[st];
	hea[st]=e;
	e++;
}
int dij(int st,int ed)
{
	//vis[st][0]=1;
	dis[st][0]=0;
	ans[st][0]=1;
	//for(int i=hea[st];i!=-1;i=eg[i].nex)
	//{
	//	dis[eg[i].v][0]=eg[i].cost;
	//	ans[eg[i].v][0]=ans[st][0];
	//}
	for(int i=1;1;i++)
	{
		int mi=0x3f3f3f;
		int f1;
		int p;
		for(int i=1;i<=n;i++)
		{
		     if(vis[i][0]==0&&dis[i][0]<mi)
		     {
		           mi=dis[i][0];
				   f1=0;
				   p=i;	
			 }
			 else if(vis[i][1]==0&&dis[i][1]<mi)
			 {
			 	mi=dis[i][1];
			 	f1=1;
			 	p=i;
			 }
		}
		if(mi==0x3f3f3f)break;
		vis[p][f1]=1;
		for(int j=hea[p];j!=-1;j=eg[j].nex)
		{
			if(dis[eg[j].v][0]>eg[j].cost+mi)
			{
				dis[eg[j].v][1]=dis[eg[j].v][0];
				ans[eg[j].v][1]=ans[eg[j].v][0];
				dis[eg[j].v][0]=eg[j].cost+mi;
				ans[eg[j].v][0]=ans[p][f1];
			}
			else if(dis[eg[j].v][0]==eg[j].cost+mi)
			{
				ans[eg[j].v][0]+=ans[p][f1];
			}
			else if(dis[eg[j].v][1]>eg[j].cost+mi)
			{
				dis[eg[j].v][1]=eg[j].cost+mi;
				ans[eg[j].v][1]=ans[p][f1];
			}
			else if(dis[eg[j].v][1]==eg[j].cost+mi)
			{
				ans[eg[j].v][1]+=ans[p][f1];
			}
		}
	}
	if(dis[ed][0]==dis[ed][1]-1)
	return ans[ed][0]+ans[ed][1];
	else return ans[ed][0];
}
int main(){
	int t;
	cin>>t;
	while(t--)
	{
		cin>>n>>m;
		memset(hea,-1,sizeof(hea));
		memset(dis,0x3f3f3f,sizeof(dis));
		memset(vis,0,sizeof(vis));
		memset(ans,0,sizeof(ans));
		e=0;
		int u,v,cost;
		for(int i=0;i<m;i++)
		{
			cin>>u>>v>>cost;
			insert(u,v,cost);
		}
		int s,f;
		cin>>s>>f;
	    cout<<dij(s,f)<<endl;;
	}
	return 0;
}