AcWing 1140. 最短网络

农夫约翰被选为他们镇的镇长!

他其中一个竞选承诺就是在镇上建立起互联网,并连接到所有的农场。

约翰已经给他的农场安排了一条高速的网络线路,他想把这条线路共享给其他农场。

约翰的农场的编号是1,其他农场的编号是 2-n.为了使花费最少,他希望用于连接所有的农场的光纤总长度尽可能短。

你将得到一份各农场之间连接距离的列表,你必须找出能连接所有农场并使所用光纤最短的方案。

#include<bits/stdc++.h>
using namespace std;
const int N=110;
int g[N][N];
int n;
int dist[N];
bool book[N];
int Prim()
{
    int res=0;
    memset(dist,0x3f,sizeof dist);
    dist[1]=0;
    for(int i=1;i<n;i++)
    {
        int t=-1;
        for(int j=1;j<=n;j++)
        if(!book[j]&&(t==-1||dist[j]<dist[t]))
         t=j;
         book[t]=1;
        for(int j=1;j<=n;j++)
        if(!book[j]) dist[j]=min(dist[j],g[t][j]);
    }
    for(int i=1;i<=n;i++) res+=dist[i];
    return res;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
     for(int j=1;j<=n;j++)
        scanf("%d",&g[i][j]);
    cout<<Prim();
    return 0;
}

 

posted @ 2021-02-04 11:22  君与  阅读(50)  评论(0编辑  收藏  举报