spfa+dij模板

#include<bits/stdc++.h>
#define ll long long
using namespace std;

ll n,m,ss,tot,head[200100],d[200100],v[200100];

struct edge
{
	ll to,w,nxt;
}e[200100];

void add(ll a,ll b,ll c)
{
	e[++tot].nxt=head[a];
	e[tot].to=b;
	e[tot].w=c;
	head[a]=tot;
}

void spfa(ll s)
{
	queue<ll> q;
	memset(d,10,sizeof d);
	v[s]=1;
	d[s]=0;
	q.push(s);
	while(!q.empty())
	{
		ll x=q.front();
		q.pop();
		v[x]=0;
		for(ll i=head[x];i;i=e[i].nxt)
		{
			ll y=e[i].to,z=e[i].w;
			if(d[y]>d[x]+z)
			{
				d[y]=d[x]+z;
				if(!v[y])
				{
					q.push(y);
					v[y]=1;
				}
			}
		}
	}
}

void dj(ll s)
{
	priority_queue< pair<ll,ll> > q;
	memset(d,10,sizeof d);
	d[s]=0;
	q.push(make_pair(0,s));
	while(!q.empty())
	{
		ll x=q.top().second;
		q.pop(); 
		if(v[x]) continue;
		v[x]=1;
		for(ll i=head[x];i;i=e[i].nxt)
		{
			ll y=e[i].to,z=e[i].w;
			if(d[y]>d[x]+z)
			{
				d[y]=d[x]+z;
				if(!v[y])
				{
					q.push(make_pair(-d[y],y));
				}
			}
		}
	}
}

int main()
{
	cin>>n>>m>>ss;
	for(ll i=1;i<=m;i++)
	{
		ll a,b,c;
		cin>>a>>b>>c;
		add(a,b,c);
	}
	dj(ss);
	for(ll i=1;i<=n;i++) cout<<d[i]<<' ';
	return 0;
}
posted @ 2024-03-01 19:42  whrwlx  阅读(9)  评论(0编辑  收藏  举报