计蒜客 难题题库 034 灌溉

•  133次 
•  13.5% 
•  1000ms 
•  65536K

到了旱季农业生产的灌溉就成了一个大问题。为了保证灌溉的顺利，某县政府决定投资为各个村之间建立灌溉管道。

输入第1行包括一个整数N，表示某县的村庄的数量。（3≤N≤100），第2行-结尾为一个N×N的矩阵,表示每个村庄之间的距离。虽然在理论上，他们是N行，每行由N个用空格分隔的数组成，实际上，他们限制在80个字符，因此，某些行会紧接着另一些行。当然，对角线将会是0，因为不会有线路从第i个村到它本身（任何两个村之间的距离都不大于100000）。

输出只有一个，为修建灌溉管道将所有村庄相连在一个灌溉系统里所需的最小管道长度。

样例1

输入：

4
0 4 9 21
4 0 8 17
9 8 0 16
21 17 16 0

输出：

28

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

const int inf = 8888888;

int main(){
    int n, i, j;
    cin >> n;
    vector<vector<int>> mp(n, vector<int>(n, 0));
    for(i = 0; i < n; ++i){
        for(j = 0; j < n; ++j){
            cin >> mp[i][j];
        }
    }
    vector<int> dis(n, inf);
    int pos = 0, res = 0, min_d = 0, count = 1;
    dis[0] = 0;
    int next_pos;
    while(count < n){
        min_d = inf;
        for(i = 0; i < n; ++i){
            if(dis[i]){
                dis[i] = min(dis[i], mp[pos][i]);
                if(min_d > dis[i]){
                    min_d = dis[i];
                    next_pos = i;
                }
            }
        }
        res += min_d;
        pos = next_pos;
        dis[pos] = 0;
        ++count;
    }
    cout << res << endl;
}