建一个超级发电站0号点,然后向每个矿井连一条边,边权就是在该点建发电站的代价,然后跑一遍kruskal,就行了
#include<bits/stdc++.h>
using namespace std;
const int N=1e5;
int n;
int fa[N];
int a[1000][1000];
int cnt=0;
struct node{
int a,b,c;
}e[N];
int find(int x)
{
if(fa[x]==x) return fa[x];
return fa[x]=find(fa[x]);
}
bool cmp(node a,node b)
{
return a.c<b.c;
}
int kru()
{
int res=0;
for(int i=1;i<=cnt;i++)
{
int u=find(e[i].a);
int v=find(e[i].b);
if(u!=v)
{
res+=e[i].c;
fa[u]=v;
}
}
return res;
}
int main()
{
cin>>n;
for(int i=0;i<=n;i++) fa[i]=i;
for(int i=1;i<=n;i++)
{
int x;
cin>>x;
e[++cnt]={0,i,x};
}
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
cin>>a[i][j];
for(int i=1;i<=n;i++)
for(int j=1;j<i;j++)
e[++cnt]={i,j,a[i][j]};
sort(e+1,e+1+cnt,cmp);
cout<<kru()<<endl;
}
为什么我不设置超级源点,而是最后把最便宜的电井加上,结果不对?