考研系列之数据结构——dijkstra算法

#define _CRT_SECURE_NO_WARNINGS
#include"iostream"
#include"cstring"
#include"cstdio"

using namespace std;
#define INF 0x7f7f7f7f

const int N = 5; //点的个数上限

int maze[N][N];//
int dis[N];//记录从源点到其他顶点当前的最短路径， 
bool vis[N];//判断是否已经加入到最短路径当中。 
int n, m;//点的个数和边的条数

void init()//初始化 
{
    memset(maze, INF, sizeof(maze));//利用初始化函数将maze数组置为无穷大 ，maze为邻接矩阵，maze[i][j]表示有向边<i,j>的权值，若没有则为INF; 
    memset(dis, INF, sizeof(dis));//将源点到其他点的距离置为无穷大 
    memset(vis, false, sizeof(vis));//将是否入队置为false，即都没有入队 
    for (int i = 0; i < N; i++)//将mazes数组中maze[i][i]的元素置为0， 顶点到自身的距离。 
    {
        maze[i][i] = 0;
    }
}

void dijkstra(int st)
{
    dis[st] = 0;
    for (int i = 0; i < n; i++)
    {
        //找到和起点距离最短的点
        int minx = INF;//初始都设为无穷大 
        int minmark;//
        for (int j = 0; j < n; j++)//n为点的个数 
        {
            if (vis[j] == false && dis[j] <= minx)//如果此顶点还没有加入最短路径并且该点长度值小于初始化的值（一般为INF) 
            {
                minx = dis[j];//更新源点到这个点的最小值， 
                minmark = j;//改变之后就要记录该数组下标 
            }
        }
        //并标记
        vis[minmark] = true;//将这个点设为已进入最短 
        //更新所有和它连接的点的距离
        for (int j = 0; j < n; j++)
        {
            if (vis[j] == false && dis[j] > dis[minmark] + maze[minmark][j])//如果 源点到该点的距离大于了有中间点加入了之后的距离 
                dis[j] = dis[minmark] + maze[minmark][j];//则更新最新距离。 
        }
    }
}


int main()
{
    while (scanf("%d %d", &n, &m) != EOF)//输入多个数据 
    {
        if (n == 0 && m == 0) break;//输入错误 
        init();//每组数据都要初始化
        for (int i = 1; i <= m; i++)
        {
            int x, y, len;
            scanf("%d %d %d", &x, &y, &len);
            if (x != y && maze[x][y] > len)
            {
                maze[y][x] = len;
                maze[x][y] = len;
            }
        }
        //以0为起点跑一次dij
        dijkstra(0);

    }
}