单源最短路径Dijkstra算法

# include<iostream>

using namespace std;

#define inf 10000

int a[5][5]={0,10,inf,30,100,
			 inf,0,50,inf,inf,
			 inf,inf,0,inf,10,
			 inf,inf,20,0,60,
			 inf,inf,inf,inf,0};          //邻接矩阵输入 

void dijkstra(int a[5][5],int n,int dis[],int pre[])
{
	int flag[n];
	flag[0]=1;
	for(int i=1;i<n;i++)
	{
		dis[i]=a[0][i];
		flag[i]=0;
		pre[i]=0;
	}            //dis[i]表示从初始节到i节点的最短路径长度，flag[i]=1表示i节点访问过，pre[i]为路径中节点i的上一个节点 
	for(int i=0;i<n;i++)
	{
		int min=inf;
		int u;
		for(int j=0;j<n;j++)
			if(dis[j]<min && !flag[j])
				{min=dis[j];u=j;}   //找dis最小边长且该边指向的节点未访问过 
			flag[u]=1;
		for(int j=0;j<n;j++)
		{
			if(!flag[j] && a[u][j]<inf)          //节点u和j之间有通路 
			{
				if(a[u][j]+dis[u]<dis[j])     // 若初始节点经过节点u到达j的长度小于直接到达j的长度 
				{
					dis[j]=a[u][j]+dis[u];    // 更新dis 
					pre[j]=u;       //记录上一个节点（以便将来输出具体路径） 
				}
			}
		}
		/*for(int i=1;i<5;i++){cout<<dis[i]<<"  ";
		}cout<<endl;*/
	}
}

int main()
{
	int dis[5],pre[5],way[5];
	dijkstra(a,5,dis,pre);
	findway(4,5,way,pre);
	cout<<"distance:"<<endl;
	for(int i=1;i<5;i++)
		cout<<dis[i]<<"  ";
	cout<<endl;
}