dijkstra最短路演算法模板(雙源)

guozuofeng發表於2018-04-16
#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;
}

相關文章