(模板)最短路徑

SPFA:
適用於存在負邊權的情況
思想是遍歷每個點,鬆弛它的出邊
時間複雜度O(nm)

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6,inf=INT_MAX;
int head[maxn],dis[maxn],cnt,n,m,s;
bool vis[maxn];
struct Edge
{
	int next,to,w;
}edge[maxn];
void add(int u,int v,int w)
{
	edge[++cnt].next=head[u];
	edge[cnt].w=w;
	edge[cnt].to=v;
	head[u]=cnt;
}
queue<int> q;
void spfa(int s)
{
	for(int i=1;i<=n;++i)
	dis[i]=inf;
	int u,v;
	q.push(s);
	dis[s]=0;
	vis[s]=true;
	while(!q.empty())
	{
		u=q.front();
		q.pop();
		vis[u]=false;
		for(int i=head[u];i;i=edge[i].next)
		{
			v=edge[i].to;
			if(dis[v]>dis[u]+edge[i].w)
			{
				dis[v]=dis[u]+edge[i].w;
				if(!vis[v])
				{
					vis[v]=true;
					q.push(v);
				}
			}
		}
	}
}
int main()
{
	cin>>n>>m>>s;
	for(int i=0;i<m;++i)
	{
		int u,v,w;
		cin>>u>>v>>w;
		add(u,v,w);
	}
	spfa(s);
	for(int i=1;i<=n;++i)
	cout<<dis[i]<<' ';
	cout<<endl;
	return 0;
}

DIJKSTRA:
適用於不存在負邊權的情況
思想是貪心,每次從最短距離中選取一個點加入最短距離已確定點集
時間複雜度O(mlogn)

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6,inf=INT_MAX;
int head[maxn],dis[maxn],cnt,n,m,s;
bool vis[maxn];
struct Edge
{
	int next,to,w;
}edge[maxn];
struct node
{
	int pos,d;	
	bool operator <(const node& x) const
	{
		return x.d<d;
	}
};
void add(int u,int v,int w)
{
	edge[++cnt].next=head[u];
	edge[cnt].w=w;
	edge[cnt].to=v;
	head[u]=cnt;
}
void dijkstra(int s)
{
	for(int i=1;i<=n;++i)
	dis[i]=inf;
	priority_queue<node> q;
	dis[s]=0;
	q.push(node{s,0});
	while(!q.empty())
	{
		node now=q.top();
		q.pop();
		int u=now.pos;
		if(vis[u]) continue;
		vis[u]=true;
		for(int i=head[u];i;i=edge[i].next)
		{
			int v=edge[i].to;
			if(dis[v]>dis[u]+edge[i].w)
			{
				dis[v]=dis[u]+edge[i].w;
				if(!vis[v])
				{
					q.push(node{v,dis[v]});
				}
			}
		}
	}
}
int main()
{
	cin>>n>>m>>s;
	for(int i=0;i<m;++i)
	{
		int u,v,w;
		cin>>u>>v>>w;
		add(u,v,w);
	}
	dijkstra(s);
	for(int i=1;i<=n;++i)
	cout<<dis[i]<<' ';
	cout<<endl;
	return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章