Floyd 最短路径算法

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

#include<iostream>
#include<vector>
using namespace std;
#define MAX 10000;
void printArr(vector<vector<int> > &arr){
    for(int i=0; i<arr.size(); ++i){
        for(int j=0; j<arr[i].size(); ++j){
            cout<<arr[i][j]<<" ";
        }
        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;
        }
    }
    //算法开始
    for(int k=0; k<N; ++k){
        for(int i=0; i<N; ++i){
            for(int j=0; j<N; ++j){
                if(arr[i][j]>arr[i][k]+arr[k][j])arr[i][j]=arr[i][k]+arr[k][j];
            }
        }
    }
    cout<<endl;
    //输出结果为一个二维矩阵，表示任意两点之间的最短路径值
    printArr(arr);
    return 0;
}

 

 测试用例：

4
0 2 6 4
-1 0 3 -1
7 -1 0 1
5 -1 12 0