最近重学C++算法,树与图的最小路径算法的代码学习。
常用的数据结构书上讲两种最短路径方法分别是(朴素迪杰斯特拉算法(Dijkstra)和弗洛伊德(Floyd)算法)。
我也暂时只准备记录更新这两种数据结构方法。
看题目:
给定一个n个点m条边的有向图,图中可能存在重边和自环,所有边权均为正值。
请你求出1号点到n号点的最短距离,如果无法从1号点走到n号点,则输出-1。
#include <bits/stdc++.h>
using namespace std;
const int N=510;
int g[N][N]; //邻接矩阵
int dist[N]; //距离
bool st[N]; //是否使用
int n,m;
//迪杰斯特拉算法
int dijkstra(int x){
memset(dist,0x3f,sizeof(dist));
dist[x]=0;
for(int i=1;i<=n;i++){ //n条边
//找dist数组中最小值的下标
int t=-1;
for(int j=1;j<=n;j++) //点从下标1开始
if(!st[j]&&(t==-1||dist[t]>dist[j]))
t=j;
for(int j=1;j<=n;j++)
dist[j]=min(dist[j],dist[t]+g[t][j]);//dist[t]代表1到t的距离
st[t]=true;
}
if(dist[n]==0x3f3f3f3f) return -1;
else dist[n];
}
int main(){
memset(g,0x3f,sizeof(g));
int a,b,v; //a->b边的值为v
cin>>n>>m;
while(m--){
cin>>a>>b>>v;
g[a][b]=min(g[a][b],v);
}
//题目是求1从开始到n点
int x=1;
printf("%d\n",dijkstra(x));
return 0;
}