最短路径dij算法
遍历与当前节点最近且未被访问过的点,逐渐更新距离,最后求得最短路径。
需要的三个数组,标记数组visit用来标记是否走过,距离数组dis[j]用来记录初始节点到当前节点j的最短距离,矩阵二维数组mp存储结点及边权值。
首先将visit全部初始化为0,dis全部初始化为INF(最大值),mp也是INF。
然后按要求输入初始化mp数组。然后标记初始节点的距离为0,即dis[start]=0;然后就双重循环,每一次定一个点,然后初始mn定为无穷大,k用来标记这个最短距离的点的下标。然后进入循环,看哪个没有被标记过,并且距离小于mn,就更新mn,并且用k标记。
循环结束后,看是否将全部循环走完却没更新k值,如果是,代表当前结点没有更短的路径了,就结束整个大循环。此时标记k点为已选择走过的点。//前面是在找距离当前节点最近的点。
然后开启新的循环,用k这个点来更新其他点的最短距离。
看例题题目与代码:
题目描述
有一个 nn 个点 mm 条边的无向图,请求出从 ss 到 tt 的最短路长度。
输入格式
第一行四个正整数 n,m,s,tn,m,s,t。 接下来 mm 行,每行三个正整数 u,v,wu,v,w,表示一条连接 u,vu,v,长为 ww 的边。
输出格式
输出一行一个整数,表示答案。
输入输出样例
输入 #1复制
7 11 5 4
2 4 2
1 4 3
7 2 2
3 4 3
5 7 5
7 3 3
6 1 1
6 3 4
2 4 3
5 6 3
7 2 1
输出 #1复制
7
代码:
#include <bits/stdc++.h>
using namespace std;
int mp[2505][2505];
int dis[2505];//存储距离
int n,m,s,t,u,v,w;
int visit[2505];//标记数组,这个点有没有被遍历过
const int INF=0x3f3f3f3f;
void dij()
{
dis[s]=0;
for(int i=1;i<=n;i++){
int mn=INF,k=0;
for(int j=1;j<=n;j++){
if(visit[j]==0&&dis[j]<mn)
{
mn=dis[j];
k=j;
}
}
if(k==0)
break;
visit[k]=1;
for(int j=1;j<=n;j++)
if(visit[j]==0&&dis[k]+mp[k][j]<dis[j]&&mp[k][j]!=INF)
dis[j]=dis[k]+mp[k][j];
}
}
int main()
{
cin>>n>>m>>s>>t;
for(int i=0;i<=n;i++)
{
dis[i]=INF;
}
for(int i=0;i<=n;i++)
for(int j=0;j<=n;j++)
mp[i][j]=INF;
for(int i=0;i<m;i++)
{
cin>>u>>v>>w;
mp[u][v]=mp[v][u]=w;
}
dij();
cout<<dis[t];
return 0;
}