1340: [USACO 3.1.1]最短网络

1340: [USACO 3.1.1]最短网络
时间限制: 1 Sec 内存限制: 64 MB
提交: 42 解决: 13
[提交][状态][讨论版]
题目描述
农民约翰被选为他们镇的镇长！他其中一个竞选承诺就是在镇上建立起互联网，并连接到所有的农场。当然，他需要你的帮助。约翰已经给他的农场安排了一条高速的网络线路，他想把这条线路共享给其他农场。为了用最小的消费，他想铺设最短的光纤去连接所有的农场。你将得到一份各农场之间连接费用的列表，你必须找出能连接所有农场并所用光纤最短的方案。每两个农场间的距离不会超过100000

输入
第一行： 农场的个数，N（3<=N<=100）。 第二行..结尾: 后来的行包含了一个N*N的矩阵,表示每个农场之间的距离。理论上，他们是N行，每行由N个用空格分隔的数组成，实际上，他们限制在80个字符，因此，某些行会紧接着另一些行。当然，对角线将会是0，因为不会有线路从第i个农场到它本身。

输出
只有一个输出，其中包含连接到每个农场的光纤的最小长度。

样例输入
4
0 4 9 21
4 0 8 17
9 8 0 16
21 17 16 0
样例输出
28

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 109;
const int infinitys = 99999999;
void inMap(int map[N][N], int lengths) {
    for (int i = 1; i <= lengths; i ++) {
        for (int j = 1; j <= lengths; j ++) {
            cin >> map[i][j];
        }
    }
}
int main(int argc, char *argv[])
{
    int sums,j,map[N][N], n, dis[N], counts, mins;
    while (cin >> n) {
        int book[N] = {0};
        inMap(map, n);
        sums = 0;
        for (int i = 1; i <= n; i ++) {
            dis[i] = map[1][i];
        }

        book[1] = 1;    //  1号顶点入树 
        counts = 1;     // 记录生成树顶点个数
        while (counts < n) {
            mins = infinitys;
            for (int i = 1; i <= n; i++) {
                if (book[i] == 0 && dis[i] < mins) {
                    mins = dis[i],j=i;
                }
            }
            book[j] = 1, counts ++, sums = sums + dis[j];

            for (int k = 1; k <= n; k ++) {
                if (book[k] == 0 && dis[k] > map[j][k]) {
                    dis[k] = map[j][k];
                }
            } 
        }
        cout << sums << endl; 
    }
    return 0;
}