计算机算法分析与设计（16）---Dijkstra算法(含C++代码)

文章目录

• 一、知识概述
• 二、代码编写

---

一、知识概述

1.1 算法描述

1.2 例题分析

二、代码编写

#include <iostream>
#include <algorithm>
using namespace std;

#define INF 9999999  //定义不可达，即无穷大 
#define MAXN 200     // 最大顶点数

//low最短距离，visit访问标记
int begin_idx, end_idx, n, k, map[MAXN][MAXN], low[MAXN], visit[MAXN]; 

void dijkstra()
{
    int m_len, index;
    for (int i = 0; i < n; i++)
    {
        low[i] = map[begin_idx][i]; //初始化low，表示从源点到其他点的最短距离 
    }
    for (int i = 0; i < n; i++)
    {
        m_len = INF;
        index = i;
        for (int j = 0; j < n; j++)
        {   //查找最短未访问距离
            if (low[j] < m_len && !visit[j])
            {
                m_len = low[j];
                index = j;
            }
        }
        visit[index] = true;
        
        for (int j = 0; j < n; j++)
        {
            int step_len = m_len + map[index][j];
            if (step_len < low[j])
            {   //是否更新距离
                low[j] = step_len;
                visit[j] = false;
            }
        }
    }
    cout << "最短距离是：" << endl;
    cout << low[end_idx] << endl;
}

int main()
{
    int a, b, len;
    
    cout<<"请输入顶点数："<< endl; 
	cin >> n;            // 顶点数
	
    cout<<"请输入边数："<< endl;
	cin >> k;            // 边数
	
	cout<<"请输入要查询的开始和结束下标："<< endl;
    cin >> begin_idx >> end_idx; // 始末下标
    
	fill(low, low + MAXN, false);     //fill是填充数组值为false 
    fill(visit, visit + MAXN, false); //fill是填充数组值为false
    
    for (int i = 0; i < MAXN; i++)
    {
        fill(map[i], map[i] + MAXN, INF); //先填充两顶点间距离为无穷大 
    }
    
	visit[begin_idx] = true;         //开始结点被访问 
    
    cout << "请输入两顶点及两顶点间的距离：" << endl; 
	for (int i = 0; i < k; i++)
    {
        cin >> a >> b >> len; //输入边的值 
        map[a][b] = map[b][a] = len;
    }
    
    dijkstra();
    return 0;
}