洛谷 P1546 最短网络 Agri-Net

题目背景

农民约翰被选为他们镇的镇长！他其中一个竞选承诺就是在镇上建立起互联网，并连接到所有的农场。当然，他需要你的帮助。

题目描述

约翰已经给他的农场安排了一条高速的网络线路，他想把这条线路共享给其他农场。为了用最小的消费，他想铺设最短的光纤去连接所有的农场。

你将得到一份各农场之间连接费用的列表，你必须找出能连接所有农场并所用光纤最短的方案。每两个农场间的距离不会超过100000

输入输出格式

输入格式：

 

第一行： 农场的个数，N（3<=N<=100）。

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

 

输出格式：

 

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

 

输入输出样例

输入样例#1： 复制

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

输出样例#1： 复制

28

#include<stdio.h>
#define N 120
int e[N][N],book[N],dis[N];
int main()
{
    int n,count=0,sum=0,i,j,u,v,min,inf=99999999;
    scanf("%d",&n);
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
            scanf("%d",&e[i][j]);
    for(i=0;i<n;i++)
        book[i]=0;
    book[0]=1;
    for(i=0;i<n;i++)
    dis[i]=e[0][i];
    count++;
    while(count<n)
    {
        min=inf;
        for(i=0;i<n;i++)
            if(book[i]==0&&dis[i]<min)
            {
                u=i;
                min=dis[i];
            }
        book[u]=1;
        sum+=dis[u];
        count++;
        for(v=0;v<n;v++)
        {
            if(book[v]==0&&dis[v]>e[u][v])
                dis[v]=e[u][v];
        }
    }
    printf("%d\n",sum);
    return 0;
}