Digkstra算法用于求某点到其他点的最短路径。
无向图的最短路径
每次找出没有访问过的点中,d[i]最小的点。然后对 i 连接的点进行判断若符合d[i]>d[k]+mp[k][i],则替换d[i]=d[k]+mp[k][i]。将每一个点都走一遍后得到最短路径。
while(1)也可替换成n次的for循环。
#include<bits/stdc++.h>
using namespace std;
int n,m,s,t;
bool vis[510];
int mp[510][510];
int d[510];
int inf=0x3f3f3f3f;
void dij(int x)
{
memset(vis,0,sizeof vis);
memset(d,0x3f3f3f3f,sizeof d);
d[x]=0;
while(1)
{
int k=-1;///imp
for(int i=1;i<=n;i++)
{
if(!vis[i]&&(k==-1 || d[i]<d[k])) k=i;
}
if(k==-1) break;
vis[k]=1;///imp
for(int i=1;i<=n;i++)
{
if(d[i]>d[k]+mp[k][i]) d[i]=d[k]+mp[k][i];
}
}
}
int main()
{
cin>>n>>m>>s>>t;
memset(mp,0x3f3f3f3f,sizeof mp);
for(int i=0;i<m;i++)
{
int u,v,p;
cin>>u>>v>>p;
mp[u][v]=p;
mp[v][u]=p;
}
dij(s);
cout<<d[t];
return 0;
}
该例子用于求s到t的最短路径。
求环
当要求s到s(环)的最短路径时,需要转化成求 s->i + i->s 的最短路径( i 为与s直接相连的点),i->s为两点直接相连的边,将该边忽略然后求出s->i的最短距离,相加后取最小值即可。
#include<bits/stdc++.h>
using namespace std;
int n,m,s,t;
bool vis[510];
int mp[510][510];
int d[510];
int inf=0x3f3f3f3f;
void dij(int u,int v)
{
memset(vis,0,sizeof vis);
memset(d,0x3f3f3f3f,sizeof d);
d[u]=0;
while(1)
{
int k=-1;///imp
for(int i=1;i<=n;i++)
{
if(!vis[i]&&(k==-1 || d[i]<d[k])) k=i;
}
if(k==-1) break;
vis[k]=1;///imp
for(int i=1;i<=n;i++)
{
if(k==u&&i==v) continue; // 跳过 s->i 直接相连边,而 i->s边仍可以走。
if(d[i]>d[k]+mp[k][i]) d[i]=d[k]+mp[k][i];
}
}
}
int main()
{
cin>>n>>m>>s>>t;
memset(mp,0x3f3f3f3f,sizeof mp);
for(int i=0;i<m;i++)
{
int u,v,p;
cin>>u>>v>>p;
mp[u][v]=p;
mp[v][u]=p;
}
int ans=0x3f3f3f3f;
for(int i=1;i<=n;i++)
{
if(i==s || mp[s][i]==inf) continue;//判断i是否直连s
dij(s,i);
if(ans>d[i]+mp[i][s]) ans=d[i]+mp[i][s];
}
cout<<ans;
return 0;
}