dijkstra最短路算法模板(双源)

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int vis[1005],dis[1005],cost[1005][1005];//vis为标记跳板点,dis为每个点与起点的距离,cost为记录点与点的权值,
long long n,m,k;
long long t1,t2;//定义起始点和终点为全局变量。
void dij(int s,int e)
{
	for(int i=1;i<=n;i++)
	{
		cost[i][i]=0;
		dis[i]=cost[s][i];//t1表示起始点,t2表示终点,
	}
	memset(vis,0,sizeof(vis));//初始化每个点为0,如果充当跳板则赋值为1
	vis[s]=1;//标记已充当过跳板的点,刚开始时,第一个起点为跳板点
	for(int i=1;i<n;i++)
	{
		int k=1e9-1;//设置k为很大很大的一个数字
		for(int j=1;j<=n;j++)
		{
			if(!vis[j] && dis[j]<k)//找出距离跳板最近的点并且这个点没有充当过跳板
			{
				k=dis[j]; //更新k的值,找出距离当前跳板最近的点
				s=j;//记录当前跳板
			}
		}
		vis[s]=1;//标记当前跳板,下次寻找过程中,不在使用这个跳板
		for(int i=1;i<=n;i++ )
		{
			if(cost[s][i]<0x3f3f3f3f)
			dis[i]=min(dis[i],dis[s]+cost[s][i]);//当前跳板为s点,判断原来每个点与起点的距离最近,
			                                     //还是起点通过跳板与每个点的距离哪儿个更近,这个过程称为松弛
		}
		if(s==e) return ;
	}
}
int main()
{
	cin>>m>>n;
	int a,b,c,k,t,sum=0.0;
	memset(cost,0x3f,sizeof(cost));
	for(int i=1;i<=m;i++)
	{
		cin>>a>>b>>c;
		cost[a][b]=min(cost[a][b],c);//无方图,a点到b点的距离,为c;
		cost[b][a]=min(cost[b][a],c);//无方图,b点到a点的距离,为c;
	}
	cin>>a>>b;
	if(a>b) swap(a,b);//起点为a,终点为b,a到b和b到a的权值是一样的!
	dij(a,b);
	cout<<dis[b]<<endl;
}

猜你喜欢

转载自blog.csdn.net/guozuofeng/article/details/79966935