数据结构——图的最短路径

图的最短路径

#include<iostream>
#include<vector>
using namespace std;
void dijkstra(const int &beg,//出发点
              const vector<vector<int> > &adjmap,//邻接矩阵，通过传引用避免拷贝
              vector<int> &dist,//出发点到各点的最短路径长度
              vector<int> &path)//路径上到达该点的前一个点
//负边被认作不联通
//福利：这个函数没有用任何全局量，可以直接复制！
{
    const int &NODE=adjmap.size();//用邻接矩阵的大小传递顶点个数，减少参数传递
    dist.assign(NODE,-1);//初始化距离为未知
    path.assign(NODE,-1);//初始化路径为未知
    vector<bool> flag(NODE,0);//标志数组，判断是否处理过
    dist[beg]=0;//出发点到自身路径长度为0
    while(1)
    {
        int v=-1;//初始化为未知
        for(int i=0; i!=NODE; ++i)
            if(!flag[i]&&dist[i]>=0)//寻找未被处理过且
                if(v<0||dist[i]<dist[v])//距离最小的点
                    v=i;
        if(v<0)return;//所有联通的点都被处理过
        flag[v]=1;//标记
        for(int i=0; i!=NODE; ++i)
            if(adjmap[v][i]>=0)//有联通路径且
                if(dist[i]<0||dist[v]+adjmap[v][i]<dist[i])//不满足三角不等式
                {
                    dist[i]=dist[v]+adjmap[v][i];//更新
                    path[i]=v;//记录路径
                }
    }
}
int main()
{
    int n_num,e_num,beg;//含义见下
    cout<<"输入点数、边数、出发点：";
    cin>>n_num>>e_num>>beg;
    vector<vector<int> > adjmap(n_num,vector<int>(n_num,-1));//默认初始化邻接矩阵
    for(int i=0,p,q; i!=e_num; ++i)
    {
        cout<<"输入第"<<i+1<<"条边的起点、终点、长度（负值代表不联通）：";
        cin>>p>>q;
        cin>>adjmap[p][q];
    }
    vector<int> dist,path;//用于接收最短路径长度及路径各点
    dijkstra(beg,adjmap,dist,path);
    for(int i=0; i!=n_num; ++i)
    {
        cout<<beg<<"到"<<i<<"的最短距离为"<<dist[i]<<"，反向打印路径：";
        for(int w=i; path[w]>=0; w=path[w])
            cout<<w<<"<-";
        cout<<beg<<'\n';
    }
}