AcWing1146新的開始
思路:
本來是跑一邊最小生成樹再加上發電站的最小值,但是wa了。
後來可以想到如果發電站的值很小,建設電網的費用很大,就可以建多個發電站。所以我們將圖轉化成n+1個點的圖,設一個虛擬的點,他到其他點的權值爲建發電站的費用。
代碼:
#include<bits/stdc++.h>
#define pii pair<int,int>
#define ll long long
#define cl(x) memset(x,0,sizeof(x))
const int N=1e6+10;
const int mod=1e7+9;
const int maxn=0x3f3f3f3f;
const int minn=0xc0c0c0c0;
const int inf=99999999;
using namespace std;
struct edge
{
int u,v,w;
}a[N];
int fa[400],maze[400][400];
int cmp(edge x,edge y)
{
return x.w<y.w;
}
int find(int x)
{
if(fa[x]!=x)
fa[x]=find(fa[x]);
return fa[x];
}
int judge(int x,int y)
{
int t1=find(x),t2=find(y);
if(t1!=t2)
{
fa[t1]=t2;
return 1;
}
return 0;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
int n,i,j,len=0;
cin>>n;
for(i=1;i<=n;i++)
{
fa[i]=i;
int k;
cin>>k;
a[++len].u=n+1;
a[len].v=i;
a[len].w=k;
}
for(i=1;i<=n;i++)
for(j=1;j<=n;j++)
cin>>maze[i][j];
for(i=1;i<=n;i++)
for(j=i+1;j<=n;j++)
{
a[++len].u=i;
a[len].v=j;
a[len].w=maze[i][j];
}
sort(a+1,a+len+1,cmp);
int ans=0,count=0;
for(i=1;i<=len;i++)
{
if(judge(a[i].u,a[i].v))
{
count++;
ans+=a[i].w;
}
if(count==n)
break;
}
cout<<ans<<endl;
return 0;
}