算法思路:
1 创建一个set,这个set可以使用bool数组表示,如果是真标明已经选择了这个点,否则还没选择。
2 创建一个数组保存路径,初始化为最大值,并初始化源点是0,这样确保我们第一个选择的点就是这个源点
3 循环寻找最短路径的点,设set数组这个点为真,表示已经选择,更新所有路径的最短距离值。
注意:
当源点到其他点没有路径的时候会容易出bug。
两个处理方法:
1 Geeks网站的: 没有路径的点也选择上,这样循环所有点,不过不会更新路径值,因为路径值还是最大值。
2 没有可选的路径的时候,直接break掉,不需要搜索了。我这里的做法。
差别不大,不过是个细节,值得指出,处理好,否则肯定会出问题的。
原网址:http://www.geeksforgeeks.org/greedy-algorithms-set-6-dijkstras-shortest-path-algorithm/
#pragma once
#include <stdio.h>
#include <algorithm>
class Dijsktra
{
const static int Vs = 9;
const static int MAX_INT = -((1<<31)+1);
int dist[Vs];
bool sptSet[Vs];
int graph[Vs][Vs];
int minDist()
{
int minD = MAX_INT, minI = -1;
for (int i = 0; i < Vs; i++)
{
if (sptSet[i] == false && dist[i] < minD)
{
minD = dist[i];
minI = i;
}
}
return minI;
}
void getShortest(int src)
{
std::fill_n(dist, Vs, MAX_INT);
std::fill(sptSet, sptSet+Vs, false);
dist[src] = 0;
for (int i = 0; i < Vs-1; i++)//可以改成Vs-1,最后一个节点不用循环
{
int id = minDist();
if (-1 == id) break;//剩下都是没有路径的点
sptSet[id] = true;
for (int k = 0; k < Vs; k++)
{
if ( !sptSet[k] && graph[id][k] && dist[id] != MAX_INT
&& dist[id] + graph[id][k] < dist[k])
{
dist[k] = dist[id] + graph[id][k];
}
}
}
}
void printDists()
{
puts("Vertex \t\t Distance From Source");
for (int i = 0; i < Vs; i++)
{
printf(" %d \t\t %d\n", i, dist[i]);
}
}
void init()
{
int gra[Vs][Vs] = {
{0, 4, 0, 0, 0, 0, 0, 8, 0},
{4, 0, 8, 0, 0, 0, 0, 11, 0},
{0, 8, 0, 7, 0, 4, 0, 0, 2},
{0, 0, 7, 0, 9, 14, 0, 0, 0},
{0, 0, 0, 9, 0, 10, 0, 0, 0},
{0, 0, 4, 0, 10, 0, 2, 0, 0},
{0, 0, 0, 14, 0, 2, 0, 1, 6},
{8, 11, 0, 0, 0, 0, 1, 0, 7},
{0, 0, 2, 0, 0, 0, 6, 7, 0}
};
for (int i = 0; i < Vs; i++)
{
for (int k = 0; k < Vs; k++)
{
graph[i][k] = gra[i][k];
}
}
}
public:
Dijsktra(int v = 0)
{
init();
getShortest(v);
printDists();
}
};