C++ dijkstra简易模板

#include
#include
#include
#include
using namespace std;
struct edge{
	int to;
	int next;
	int w;
}edge[500010];
int cnt=0;
int head[100010];
int d[100010];
int vis[100010];
int n,m,u,v,w,s;
priority_queue<pair<int,int> >q;

inline void Add(int u,int v,int w){
	cnt++;
	edge[cnt].to=v;
	edge[cnt].w=w;
	edge[cnt].next=head[u];
	head[u]=cnt;
}
inline void DJ(int s){
	memset(d,0x3f,sizeof(d));
	d[s]=0;
	q.push(make_pair(0,s));
	while(!q.empty())
	{
		int x=q.top().second;
		q.pop();
		if(vis[x])
		continue;
		vis[x]=1;
		for(int i=head[x];i;i=edge[i].next)
		{
			int y=edge[i].to;
			int z=edge[i].w;
			if(d[y]>d[x]+z)
			{
				d[y]=d[x]+z;
				if(!vis[y])
				{
					q.push(make_pair(-d[y],y));
				}
			}
		}
	}
}
int main()
{
	cin>>n>>m>>s;
	for(int i=1;i<=m;i++){
		cin>>u>>v>>w;
		Add(u,v,w);
	}
	DJ(s);
	for(int i=1;i<=n;i++){
		cout<<d[i]<<" ";
	}
} 

你可能感兴趣的:(数据结构)