dijkstra单源最短路课设代码详解

/*朴素版dijkstra 时间复杂度最坏情况为O(n^2) 
dijkstra算法采用的是一种贪心的策略，每次遍历到始点距离最近且未访问过的顶点的邻接节点，直到扩展到终点为止。 
用一个dist数组保存源点到其余各点的距离，由于要求的时最短距离，所以初始化时，dist数组应该初始化为正无穷。
用一个标记数组st记录是否找到了源点到该节点的最短距离，如果st[i]为true，则表示找到了源点到节点i的最短距离。
如果为false，说明还没有找到，初始化数组st为false。 
实现过程：（伪代码） 
1.初始化all(dist[])=0x3f3f3f3f,dist[1]=0,S为已经确定最短距离的点的集合
2.for(int i=1;i<=n;i++)
{
    1. 寻找不在S中的距离最近的点t
    2. 将t加入到s中
    3. 用t更新其他点的距离 
}
*/
//1. 朴素版dijkstra之稠密图 
#include <bits/stdc++.h>
using namespace std;
const int N=510;
int g[N][N];//稠密图用邻接矩阵 
int dist[N];//dist[i]表示i到1节点的距离
bool st[N];//标记数组。如果i节点的距离已经被确定，则标记为true 
int n,m;//n个点m条边
int dijkstra()
{
    memset(dist,0x3f,sizeof dist);
    dist[1]=0;//除1节点外，其余点设为正无穷
    for(int i=0;i<n;i++)//n次迭代，每次寻找不在s中距离最近的点t
    {
        int t=-1;//这里t的作用是为了便于更新第一个点，充当临时变量的作用。 
        for(int j=1;j<=n;j++)//循环遍历n个节点
            if(!st[j]&&(t==-1||dist[t]>dist[j]))
                t=j;
        st[t]=true;//更新这个点之后就把他标记，也就是把他加入集合S中
        for(int j=1;j<=n;j++)//以t为跳板，更新t周围与之相邻的点 
            dist[j]=min(dist[j],dist[t]+g[t][j]); 
    }
    if(dist[n]==0x3f3f3f3f)return -1;//n没有被更新，即路径不存在。 
    else return dist[n]; 
}
int main()
{
    cout<<"请输入点的个数以及边的个数"<<endl;
    cout<<"点的个数n："<<endl;
    cin>>n;
    cout<<"边的个数m: "<<endl;
    cin>>m;
    cout<<"请输入m条边：例如： a b c表示a->b边长为c"<<endl; 
    memset(g,0x3f,sizeof g);//初始化为正无穷表示两个点之间的距离还没有被更新
    for(int i=1;i<=m;i++)
    {
        int a,b,c;
        cout<<"请输入第 "<<i<<" 条边："<<endl;
        cin>>a>>b>>c;
        g[a][b]=c;//a->b长度为c 
    }
    cout<<"1->n的距离是: ";
    cout<<dijkstra()<<endl;
    return 0; 
} 

//dijkstra之堆优化O(mlogn)，主要用于稀疏图 
#include <bits/stdc++.h>
#define x first//方便写first和second，无其他含义 
#define y second
using namespace std;
const int N=510; 
int n,m;
int h[N],e[N],ne[N],w[N],idx; 
//h数组存头节点，e数组存节点编号,ne数组也就是next数组存后继节点,w数组存边权值
int dist[N];
bool st[N];
typedef pair<int,int>PII;
priority_queue<PII,vector<PII>,greater<PII>>heap;//小根堆，每次更新一次排序的情况是log的复杂度，相较于朴素版本的优化体现在这里。
void add(int a,int b,int c)// 链式前向星存图，可以理解为数组模拟单链表的尾插法。 a->b，权值为c。 
{
    e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
} 
int dijkstra()
{
    memset(dist,0x3f,sizeof dist);
    dist[1]=0;
    heap.push({0,1});//{离起点的距离，点的编号}
    while(heap.size())
    {
        auto t=heap.top();//取堆顶元素，也就是离起点最近的点
        heap.pop();
        int ver=t.y;//节点
        int distance=t.x;//离起点的距离
        if(st[ver])continue;//如果被更新过，就看下一个点
        for(int i=h[ver];~i;i=ne[i])//遍历与ver相邻的点 
        {
            int j=e[i];//周围点为 j
            if(dist[j]>distance+w[i])
            {
                dist[j]=distance+w[i];//更新 
                heap.push({dist[j],j});//存入堆中 
            } 
        } 
    }
    if(dist[n]==0x3f3f3f3f)return -1;//不存在路径 
    else return dist[n];
}
int main()
{
    memset(h,-1,sizeof h);
    cout<<"请输入点的个数以及边的个数"<<endl;
    cout<<"点的个数n："<<endl;
    cin>>n;
    cout<<"边的个数m: "<<endl;
    cin>>m;
    cout<<"请输入m条边：例如： a b c表示a->b边长为c"<<endl; 
    for(int i=1;i<=m;i++)
    {
        int a,b,c;
        cout<<"请输入第 "<<i<<" 条边："<<endl;
        cin>>a>>b>>c;
        add(a,b,c);//a->b长度为c 
    }
    cout<<"1->n的距离是: ";
    cout<<dijkstra()<<endl;
    return 0;
}