图论-最短路问题(Dijkstra和Floyd)

Dijkstra算法

时间复杂度：O((n+m)logn)

#include <bits/stdc++.h>
#define int long long

using namespace std;

const int N=1e5+5,M=2e5+5,INF=0x3f3f3f3f;

typedef pair<int,int> PII;

int h[M],e[M],ne[M],va[M],idx;

void add(int a,int b,int c)
{
	e[idx]=b;
	va[idx]=c;
	ne[idx]=h[a];
	h[a]=idx++;
}

int n,m,x,y,z;
int d[N];
bool vis[N];
priority_queue<PII,vector<PII>,greater<PII>> q;

void dijkstra()
{
	memset(d,INF,sizeof d);
	d[1]=0;
	q.push({0,1});
	while(q.size())
	{
		int x=q.top().second;
		q.pop();
		if(vis[x]) continue;
		vis[x]=true;
		for(int i=h[x];~i;i=ne[i])
		{
			int y=e[i],z=va[i];
			if(d[y]>d[x]+z)
			{
				d[y]=d[x]+z;
				q.push({d[y],y});
			}
		}
	}	
}

signed main()
{
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	cin>>n>>m;
	memset(h,-1,sizeof h);
	for(int i=1;i<=m;i++)
	{
		cin>>x>>y>>z;
		add(x,y,z);
		add(y,x,z);
	}
	
	dijkstra();
	
	for(int i=2;i<=n;i++) cout<<d[i]<<" ";
	
	return 0;
}

例题：Dijkstra? - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

题解代码：

#include <bits/stdc++.h>
#define int long long

using namespace std;

const int N=1e5+5,M=2e5+5,INF=0x3f3f3f3f3f3f3f3f;

typedef pair<int,int> PII;

int h[M],e[M],ne[M],va[M],idx;

void add(int a,int b,int c)
{
	e[idx]=b;
	va[idx]=c;
	ne[idx]=h[a];
	h[a]=idx++;
}

int n,m,x,y,z;
int d[N],fa[N];
bool vis[N];
priority_queue<PII,vector<PII>,greater<PII>> q;

void dijkstra()
{
	memset(d,INF,sizeof d);
	d[1]=0;
	q.push({0,1});
	while(q.size())
	{
		int x=q.top().second;
		q.pop();
		if(vis[x]) continue;
		vis[x]=true;
		for(int i=h[x];~i;i=ne[i])
		{
			int y=e[i],z=va[i];
			if(d[y]>d[x]+z)
			{
				d[y]=d[x]+z;
				fa[y]=x;
				q.push({d[y],y});
			}
		}
	}	
}

signed main()
{
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	cin>>n>>m;
	memset(h,-1,sizeof h);
	memset(fa,-1,sizeof fa);
	for(int i=1;i<=m;i++)
	{
		cin>>x>>y>>z;
		add(x,y,z);
		add(y,x,z);
	}
	
	dijkstra();
	
	if(d[n]==INF)
	{
		cout<<-1;
		return 0;
	}
	
	vector<int> ans;
	for(int i=fa[n];~i;i=fa[i]) ans.push_back(i);
	reverse(ans.begin(),ans.end());
	for(auto u:ans) cout<<u<<" ";
	cout<<n;
	
	return 0;
}

Floyd算法

时间复杂度：O(n^3)

#include <bits/stdc++.h>

using namespace std;

const int N=3e3+5,INF=0x3f3f3f3f;

int n,m,d[N][N];

signed main()
{
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	cin>>n>>m;
	memset(d,INF,sizeof d);
	for(int i=1;i<=n;i++) d[i][i]=0;
	
	for(int i=1,x,y,z;i<=m;i++)
	{
		cin>>x>>y>>z;
		d[x][y]=min(d[x][y],z);
	}
	
	for(int k=1;k<=n;k++)
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
				d[i][j]=min(d[i][j],d[i][k]+d[k][j]);
				
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++) cout<<d[i][j]<<" ";
		cout<<endl;
	}
	
	return 0;
}

例题：https://www.luogu.com.cn/problem/B3647

题解代码：

#include <bits/stdc++.h>

using namespace std;

const int N=505,INF=0x3f3f3f3f;

int n,m,d[N][N];

signed main()
{
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	cin>>n>>m;
	memset(d,INF,sizeof d);
	for(int i=1;i<=n;i++) d[i][i]=0;
	
	for(int i=1,x,y,z;i<=m;i++)
	{
		cin>>x>>y>>z;
		d[y][x]=d[x][y]=min(d[x][y],z);
	}
	
	for(int k=1;k<=n;k++)
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
				d[i][j]=min(d[i][j],d[i][k]+d[k][j]);
	
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++) cout<<d[i][j]<<" ";
		cout<<endl;
	}
	
	return 0;
}