计蒜客 实验4-4灌溉 生成树模板

题目：

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

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

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

样例输入

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

样例输出

28

思路：Prim模板

#include <bits/stdc++.h>
using namespace std;
#define maxv 100+5
#define INF 1<<29
int cost[maxv][maxv];
bool used[maxv];
int d[maxv];//当前时刻到第i个点的最短距离
int N;

int prim(int st)
{
    fill(used,used+N+1,false);
    fill(d,d+N+1,INF);
    d[st]=0;
    int res=0;//存结果
    while(1)
   {
       int v=-1;
    for(int i=1;i<=N;i++)
    {

        if(!used[i]&&(v==-1||d[i]<d[v]))
            v=i;
    }
    if(v==-1)
        break;
        used[v]=1;
       // cout<<v<<" "<<d[v]<<endl;
        res+=d[v];
        for(int i=1;i<=N;i++)
        {
            if(!used[i]&&d[i]>cost[v][i])
               d[i]=cost[v][i];  //和ijkstra的区别，dijkstra表示的是从起到到当前点最短距离
        }
   }
   return res;
}
int main()
{
    scanf("%d",&N);
    for(int i=1;i<=N;i++)
    for(int j=1;j<=N;j++)
        scanf("%d",&cost[i][j]);
    int ans=prim(N);
    printf("%d",ans);
    return 0;
}