dijkstra算法

朴素版

#include<bits/stdc++.h>
using namespace std;
const int N=510; 
int a,b;
int g[N][N];//邻界表 
int dist[N];//到每个点的距离 
bool st[N];//该点是否被确认 
int dijkstra()
{
	memset(dist,0x3f,sizeof dist);
	dist[1]=0;
	for(int k=1;k<=a;k++)
	{
			int t=-1;
			for(int j=1;j<=a;j++)
			{
				if(!st[j]&&(t==-1||dist[t]>dist[j]))
				{
					t=j;
				}
			}
		    st[t]=true;
			for(int j=1;j<=a;j++)
			{
				dist[j]=min(dist[j],dist[t]+g[t][j]);
			}
	}
	if(dist[a]==0x3f3f3f3f) return -1;
	return dist[a];
}
int main()
{
	memset(g,0x3f,sizeof g);
	cin>>a>>b;
	for(int k=1;k<=b;k++)
	{
		int c,d,e;
		cin>>c>>d>>e;
		g[c][d]=min(g[c][d],e);
	}
	cout<<dijkstra()<<endl;
	return 0;
}

堆优化

#include<bits/stdc++.h>
using namespace std;
const int N=150000;
int h[N],e[N],ne[N],w[N],idx;
int dist[N];
bool st[N];
int n,m;
typedef pair <int,int> PII;
void add(int a,int b,int c)
{
	e[idx]=b;
	ne[idx]=h[a];
	w[idx]=c;
	h[a]=idx++;
}
int dijkstra()
{
	memset(dist,0x3f,sizeof dist);
	dist[1]=0;
	priority_queue<PII, vector<PII>, greater<PII> > heap;
	heap.push({0,1});
	while(heap.size())
	{
		PII j = heap.top();		
		heap.pop();
		int d = j.first;
		int ver = j.second;
		if(st[ver]) continue;
		st[ver]=true;
		for(int k=h[ver];k!=-1;k=ne[k])
		{
			int j=e[k];
			if(dist[j]>d+w[k])
			{
				dist[j]=d+w[k];
				heap.push({dist[j],j});
			}
		}
	}
	if(dist[n]==0x3f3f3f3f) return -1;
	else return dist[n];
}
int main()
{
	cin>>n>>m;
	memset(h,-1,sizeof h);
	for(int k=1;k<=m;k++)
	{
		int a,b,c;
		cin>>a>>b>>c;
		add(a,b,c);
	}
	cout<<dijkstra()<<endl;
	return 0;
}