Dijkstra最短路径算法[贪心]

#include<iostream>
#include<stdio.h>
#include<limits.h>
using namespace std;

const int V=9;
//从未包含在SPT的集合T中，选取一个到S集合的最短距离的顶点
int getMinIndex(int dist[V],bool sptSet[V])
{
    int min=INT_MAX,min_index;
    for(int v=0;v<V;v++)
    {
        if(!sptSet[v]&&dist[v]<min)
            min=dist[v],min_index=v;
    }
     return min_index;
}

//打印结果

void  printSolution(int dist[],int n)
{
    printf("Vertex Distance from Source\n");
    for(int i=0;i<V;i++)
        printf("%d\t\t %d\n",i,dist[i]);
}

//source 代表顶点
void dijkstra(int graph[V][V],int source)
{
    int dist[V];// 存储结果，从源点到 i的距离
    bool sptSet[V]; // sptSet[i]=true 如果顶点i包含在SPT中

    // 初始化. 0代表不可达
    for(int i=0;i<V;i++)
    {
        dist[i]=(graph[source][i]==0)?INT_MAX:graph[source][i];
        sptSet[i]=false;
    }
     // 源点，距离总是为0. 并加入SPT
    dist[source]=0;
    sptSet[source]=true;
    
    // 迭代V-1次，因此不用计算源点了，还剩下V-1个需要计算的顶点。
    for(int count=0;count<V-1;count++)
    {
     // u，是T集合中，到S集合距离最小的点,u开始在T集合中，然后加入到S集合
        int u=getMinIndex(dist,sptSet);
        // 加入SPT中
        sptSet[u]=true;
        for(int v=0;v<V;v++)
        {
        //满足以下4个条件
        //1. 点v还没有加入到spt中
        //2. 点u和v之间可达
        //3. 点u是可达的，距离不是无穷
        //4. 经过点u之后的距离小于直接到达点v的距离
            if(!sptSet[v]&&graph[u][v]&&dist[u]!=INT_MAX&&dist[u]+graph[u][v]<dist[v])
                dist[v]=dist[u]+graph[u][v];
        }
    }
    printSolution(dist,V);
}

int main()
{
    int graph[V][V]={ { 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 }
                    };

    dijkstra(graph, 0);

    return 0;
}