最短路径——dijkstra

书上的模板,贴上来只是方便我查询。
唯一忘记说的是,只要无路可走,最短路径即为无穷大。
代码入下

#include
#include
#include
#include
#include
#define INF 1000000
#define CLR(x) memset(x,0,sizeof(x))
using namespace std;

const int maxp = 20;
int p ;
int edge[maxp][maxp];
int dist[maxp];//表示距离。(最终结果为每个点到起始点的最短距离)
int vis[maxp];//判断该点是否被标记。
int path[maxp];//记录最短路径

int dijkstra(int v0)
{
    for(int i = 0 ; i < p ; i++)
    {
        dist[i] = edge[v0][i];
        vis[i] = 0;
        if(i!=v0 && dist[i] < INF) path[i] = v0;
        else path[i] = -1;
    }
    vis[v0] = 1;//由于以下写法会忽略第一个点所以要将三个数组全部初始化为关于第一个点的值。
    //dist[v0] = 0;
    for(int i = 0 ; i < p-1 ; i++)//顶天只能再找p-1个点了。
    {
        int minn = INF;
        int u = v0;
//===========================================================在这里找到当前节点所连接的点中最短路段的点
        for(int j = 0; j < p ; j++)
        {
            if(vis[j] != 1 && dist[j] < minn)
            {
                u = j ; minn = dist[j];
            }
        }
//===========================================================
        vis[u] = 1;    //找到后将该点选取并表示以后不再选取该点。
//===========================================================
//在这一段中判断每一个点到被选取的点的距离是否长于当前节点和被选取点的距离与被选取点和现在测试的点的距离之和,若是,则更新。
//更新后表示从当前节点到被选取节点再到那个点路径更短。否则就是被选取点到该点距离最短。
//若更新,顺便更新path数组表示该点的上一个节点是被选取节点。
        for(int j = 0 ; j < p ; j++)
        {
            if(vis[j] != 1 && dist[u]+edge[u][j] < dist[j])
            {
                dist[j] = dist[u]+edge[u][j] ;
                path[j] = u;
            }
        }
//============================================================
    }

}

int main()
{
    scanf("%d",&p);
    while(1)
    {
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        if(u == -1 && v == -1 && w == -1) break;
        edge[u][v] = w;
    }
    for(int i = 0 ; i < p ; i++)
        for(int j = 0 ; j < p ; j++)
        {
            if( i == j) edge[i][j] = 0;
            else if(edge[i][j] == 0) edge[i][j] = INF;
        }
    dijkstra(0);
    int shorttest[maxp];
    for(int i = 1 ; i < p ; i++)
    {
        printf("%d\t",dist[i]);
        CLR(shorttest);
        int k = 0;
        shorttest[k] = i;
        while(path[shorttest[k]] != 0)
        {
            k++;
            shorttest[k] = path[ shorttest[k-1] ];
        }
        k++;
        shorttest[k] = 0;
        for(int j = k ; j > 0 ; j--)
        {
            printf("%d --> ",shorttest[j]);
        }
        printf("%d\n",shorttest[0]);
    }

}

/*样例如下
6
0 2 5
0 3 30
1 0 2
1 4 8
2 5 7
2 1 15
4 3 4
5 3 10
5 4 18
-1 -1 -1
*/





你可能感兴趣的:(最短路径——dijkstra)