以朴素版dijkstra(稠密图)(邻接矩阵)解
#include<bits/stdc++.h>
#define endl '\n'
using namespace std;
const int N=2510;
const int INF=0x3f3f3f3f;//以0x3f3f3f3f代表正无穷
int n,m,S,T,u,v,w;
int g[N][N];
int dist[N];
bool st[N];
void dijkstra()
{
dist[S]=0;
for(int i=0;i<n;i++)
{
int t=-1;
for(int j=1;j<=n;j++)
{
if(!st[j]&&(t==-1||dist[j]<dist[t]))//寻找一个还未确定最短距离的点
t=j;
}
st[t]=true;//更新此点的状态为已确定最短距离
for(int j=1;j<=n;j++)//用此点来更新其他点
{
dist[j]=min(dist[j],dist[t]+g[t][j]);//比较通过此点到达目标点的最短距离或者目标点原来的最短距离,取最小值
}
}
}
int main()
{
cin>>n>>m>>S>>T;
for(int i=0;i<=n;i++)
{
dist[i]=INF;
for(int j=0;j<=n;j++)
{
if(i==j)g[i][j]=0;//初始化自己到自己的距离为0
else g[i][j]=INF;//其余为正无穷
}
}
for(int i=0;i<m;i++)
{
cin>>u>>v>>w;
g[u][v]=min(g[u][v],w);//取两点之间的最小距离,来处理重边和自环
g[v][u]=min(g[v][u],w);//双向边
}
dijkstra();
cout<<dist[T];
return 0;
}
堆优化版dijkstra(稀疏图)(邻接表)解
#include<bits/stdc++.h>
using namespace std;
const int N=2510;
int n,m,s,t;
typedef pair<int,int> PII;//first为邻点,second为两点之间的边权
vector<PII> g[N];//邻接表存储图
int dist[N];//记录每个点的最小距离
bool vis[N];//判断此点状态,是否已确定最小距离
void dijkstra()
{
for(int i=0;i<=n+1;i++)
{
vis[i]=false;
dist[i]=1e9;
}
dist[s]=0;
priority_queue<PII,vector<PII>,greater<PII>>q;//q存储的是个pair
//first为此点到起点的最短距离,second为此点的编号
//用堆来维护还未确定最小距离的点中距离起点距离最小的点
//由于小根堆会优先按照第一个数据进行排序,所以要把距离放到first的位置
q.push({dist[s],s});
while(q.size())
{
int u=q.top().second;//取出此编号
q.pop();
if(vis[u])continue;//如果此编号已经被确定最小距离则为冗余备份,跳过此次更新
vis[u]=true;//更新此点状态为已确定最小距离
for(auto [v,w]:g[u])//用u的最小距离来更新u的所有临边连接的点的最小距离
{
if(dist[v]>dist[u]+w)
{
dist[v]=dist[u]+w;
q.push({dist[v],v});
}
}
}
}
int main()
{
cin>>n>>m>>s>>t;
for(int i=0;i<m;i++)
{
int u,v,w;
cin>>u>>v>>w;
g[u].push_back({v,w});
g[v].push_back({u,w});
}
dijkstra();
cout<<dist[t];
return 0;
}
由于此题边数不超过1e4所以更宜用堆优化版解答(点也不多,朴素版也能过)
其他算法补充:
#include <bits/stdc++.h>
#define endl '\n'
#define INF 0x3f3f3f3f
using namespace std;
const int N=2510;
typedef pair<int,int> pii;
vector<pii> g[N];
int dist[N];
bool vis[N];
int n,m,s,t;
void init()
{
for(int i=0;i<=n;i++)
{
vis[i]=false;
dist[i]=INF;
}
}
void dijkstra()
{
dist[s]=0;
priority_queue<pii,vector<pii>,greater<pii>> q;
q.push({dist[s],s});
while(q.size())
{
int u=q.top().second;
q.pop();
if(vis[u])continue;
vis[u]=true;
for(auto [v,w]:g[u])
{
if(dist[v]>dist[u]+w)
{
dist[v]=dist[u]+w;
q.push({dist[v],v});
}
}
}
}
void bellman()//bellman不用vis数组判断点
{
dist[s]=0;
for(int i=0;i<n;i++)
{
for(int u=1;u<=n;u++)
{
for(auto [v,w]:g[u])
{
dist[v]=min(dist[v],dist[u]+w);
}
}
}
}
void spfa()
{
dist[s]=0;
vis[s]=true;//放入队列更新状态为待松弛
queue<int>q;
q.push(s);
while(q.size())
{
int u=q.front();
q.pop();
vis[u]=false;//已松弛当前结点
for(auto [v,w]:g[u])
{
if(dist[v]>dist[u]+w)
{
dist[v]=dist[u]+w;
if(!vis[v])//如果更新结点未在队列中,则将该点加入队列
{
q.push(v);
vis[v]=true;
}
}
}
}
}
int main()
{
cin>>n>>m>>s>>t;
for(int i=0;i<m;i++)
{
int u,v,w;
cin>>u>>v>>w;
g[u].push_back({v,w});
g[v].push_back({u,w});
}
init();
spfa();
//dijkstra();
//bellman();
cout<<dist[t];
return 0;
}