Dijkstra 最短路径算法

详见：http://wiki.jikexueyuan.com/project/easy-learn-algorithm/dijkstra.html

#include<iostream>
#include<vector>
using namespace std;
#define MAX 10000
void printArr(vector<int> &arr){
    for(int i=0; i<arr.size(); ++i){
        cout<<arr[i]<<" ";
    }
    cout<<endl;
}
int main(){
    //数据读入，输入为N（表示邻接矩阵的边长），后面紧接一个维度为N的邻接矩阵，-1表示正无穷
    int N;
    cin>>N;
    vector<vector<int> > arr(N, vector<int>(N, 0));
    for(int i=0; i<N; ++i){
        for(int j=0; j<N; ++j){
            cin>>arr[i][j];
            if(arr[i][j]==-1)arr[i][j]=MAX;
        }
    }
    int startPoint=0;//设置起点为0号点，当然也可以设置为1号、2号
    vector<int> dist(N, 0);
    for(int j=0; j<N; ++j)dist[j]=arr[startPoint][j];
    vector<bool> flag(N, false);
    flag[startPoint]=true;
    int minIdx; 
    for(int k=1; k<N; ++k){
        int min=MAX;
        for(int i=0; i<N; ++i){
            if(flag[i]==false && dist[i]<min){
                min=dist[i];
                minIdx=i;
            }
        }
        flag[minIdx]=true;
        for(int j=0; j<N; ++j){
            if(dist[j]>dist[minIdx]+arr[minIdx][j])dist[j]=dist[minIdx]+arr[minIdx][j];
        }
    }
    //输出结果为startPoint（起点）到其他各点的最短路径值
    printArr(dist);
    return 0;
}

测试用例：

6
0 1 12 -1 -1 -1
-1 0 9 3 -1 -1
-1 -1 0 -1 5 -1
-1 -1 4 0 13 15
-1 -1 -1 -1 0 4
-1 -1 -1 -1 -1 0