Agri-Net ——最小生成树模板题（矩阵输入）

Posted 2022-12-12 j666

题目链接

题意：

给你一个矩阵，表示 i j 连接需要的花费，求出最小的花费

题解：

裸最小生成树板子

处理好矩阵输入即可

 

代码：

#include<iostream>
#include<stdio.h>
#include<math.h>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
const int maxn = 105;
int f[maxn];
int x[maxn],y[maxn];
int n,cnt,m;
struct node

    int u,v;
    int w;
    bool operator < (const node &a)const
    
        return w<a.w;
    
 edge[maxn*maxn];

int Find(int x)

    return x==f[x]?x:f[x]=Find(f[x]);

void add(int u,int v,int w)

    edge[cnt].u=u;
    edge[cnt].v=v;
    edge[cnt++].w=w;


int kruskal()

    int ans=0;
    for(int i=0; i<=n; i++)f[i]=i;
    sort(edge,edge+cnt);
    for(int i=0; i<cnt; i++)
    
        int x=edge[i].u;
        int y=edge[i].v;
        int fx=Find(x);
        int fy=Find(y);
        if(fx!=fy)
        
            f[fx]=fy;
            ans+=edge[i].w;
        
    
    return ans;


int main()

    while(~scanf("%d",&n))
    
        cnt=0;
        for(int i=0; i<=n; i++)f[i]=i;

        for(int i=1; i<=n; i++)for(int j=1; j<=n; j++)
        
            int x;
            scanf("%d",&x);
            add(i,j,x);
        
        printf("%d\n",kruskal());
    



    return 0;

kruskal