【BZOJ 1601】 灌水

【题目链接】

https://www.lydsy.com/JudgeOnline/problem.php?id=1601

【算法】

最小生成树

【代码】

#include<bits/stdc++.h>
using namespace std;
#define MAXN 310
 
struct Edge
{
        int x,y,w;
} e[MAXN*MAXN];
 
int i,j,w,n,ans,tot;
int s[MAXN];
 
inline bool cmp(Edge a,Edge b)
{
        return a.w < b.w;
}
inline int find(int x)
{
        if (s[x] == x) return x;
        return s[x] = find(s[x]);
}
 
int main()
{
 
        scanf("%d",&n);
        for (i = ; i <= n; i++)
        {
                scanf("%d",&w);
                e[++tot] = (Edge){,i,w};
        }
        for (i = ; i <= n; i++)
        {
                for (j = ; j <= n; j++)
                {
                        scanf("%d",&w);
                        if (i < j) e[++tot] = (Edge){i,j,w};
                }
        }
        for (i = ; i <= n; i++) s[i] = i;
        sort(e+,e+tot+,cmp);
        for (i = ; i <= tot; i++)
        {
                if (find(e[i].x) != find(e[i].y))
                {
                        s[find(e[i].x)] = find(e[i].y);
                        ans += e[i].w;
                }
        }
        printf("%d\n",ans);
 
        return ;
 
}