LOJ10066

LOJ10066 新的开始

题目描述

发展采矿业当然首先得有矿井，小 F 花了上次探险获得的千分之一的财富请人在岛上挖了 n 口矿井，但他似乎忘记考虑的矿井供电问题……

为了保证电力的供应，小 F 想到了两种办法：

1. 在这一口矿井上建立一个发电站，费用为 v（发电站的输出功率可以供给任意多个矿井）。
2. 将这口矿井与另外的已经有电力供应的矿井之间建立电网，费用为 p。

小 FF 希望身为「NewBe_One」计划首席工程师的你帮他想出一个保证所有矿井电力供应的最小花费。

输入格式

第一行一个整数 n，表示矿井总数。

第 2∼n+1 行，每行一个整数，第 i 个数vi​ 表示在第i 口矿井上建立发电站的费用。

接下来为一个n×n 的矩阵 pp，其中pi,j​ 表示在第 i 口矿井和第 j 口矿井之间建立电网的费用（数据保证有pi,j​=pj,i​，且 pi,i​=0）。

输出格式

输出仅一个整数，表示让所有矿井获得充足电能的最小花费。

样例

样例输入

4  
5  
4 
4  
3  
0 2 2 2  
2 0 3 3  
2 3 0 4  
2 3 4 0

样例输出

9

样例解释

小 FF 可以选择在 4 号矿井建立发电站然后把所有矿井都不其建立电网，总花费是3+2+2+2=9。

数据范围与提示

对于 30% 的数据：1≤n≤50；
对于 100% 的数据：1≤n≤300,0≤vi​,pi,j​≤10^5。

_______________________________________________________

最小生成树，记得建立0号点，把0号点到其他点的边长设为建电站的花费。

_______________________________________________________

#include<bits/stdc++.h>
using namespace std;
const int maxn=305;
struct edge
{
    int u,v,w;
}e[maxn*maxn];
int n,js,cnt;
long long ans;
void addage(int u,int v,int w)
{
    e[js].u=u;e[js].v=v;e[js++].w=w;
}
bool mycmp(edge a,edge b)
{
    return a.w<b.w;
}
int fa[maxn];
int find(int x)
{
    if(x==fa[x])return x;
    return fa[x]=find(fa[x]);
}
int main()
{
    scanf("%d",&n);
    for(int x,i=1;i<=n;++i)
    {
        scanf("%d",&x);
        addage(0,i,x);
    }
    for(int x,i=1;i<=n;++i)
    {
        for(int j=1;j<=n;++j)
        {
            scanf("%d",&x);
            if(i<j)addage(i,j,x);
        }
    }
    for(int i=0;i<=n;++i)fa[i]=i;
    sort(e,e+js,mycmp);
    for(int i=0;i<js;++i)
    {
        int a=find(e[i].u),b=find(e[i].v);
        if(a!=b)
        {
            fa[a]=b;
            ans+=e[i].w;
            ++cnt;
            if(cnt==n)break;
        }
    }
    cout<<ans;
    return 0;
}