(最小生成树)AcWing1146新的开始

AcWing1146新的开始

思路:

本来是跑一边最小生成树再加上发电站的最小值,但是wa了。
后来可以想到如果发电站的值很小,建设电网的费用很大,就可以建多个发电站。所以我们将图转化成n+1个点的图,设一个虚拟的点,他到其他点的权值为建发电站的费用。

代码:

#include
#define pii pair
#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;
}

你可能感兴趣的:(图论,#,最小生成树)