【JZOJ】【最小生成树】给水

$Description$

由n块草地组成了一个花园
现在要你给这个花园里的所有草地供水
你可以在第$i$块草地挖井，费用为$A[i]$
也可以从其他草地连接水管过来
但前提是那个草地要有水，费用为$F[i][j]$
现在要你求给所有草地供水
最少需要多少大洋

$Input$

第一行，一个$n$
第$2$~$n+1$行，每行一个$A[i]$
第$n+2$~$2n+1$行，每行$n$个数，代表$F[i][j]$

$Output$

一行，为最小费用

$SampleInput$

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

$SampleOutput$

9

思路

最小生成树

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int A[550],F[550],K[550],L[550][550];
int Ans,n,T;
int main()
{
	freopen("water.in","r",stdin);
	freopen("water.out","w",stdout);
	scanf("%d",&n);
	T=n+1;
	for(int i=1;i<=n;++i)
	{
		scanf("%d",&A[i]);
		L[i][T]=L[T][i]=A[i];
	}
	for(int i=1;i<=n;++i)
		for(int j=1;j<=n;++j)
			scanf("%d",&L[i][j]);
	memset(F,0x3f,sizeof(F));
	F[1]=0;//现凿一个井
	for(int i=1;i<=T;++i)
	{
		int l=0;
		for(int j=1;j<=T;++j)
			if(F[j]<F[l] && !K[j])l=j;//看目前到达所需要的费用最少的点
		Ans+=F[l],K[l]=1;
		for(int j=1;j<=T;++j)
			if(l!=j && L[l][j]<F[j])F[j]=L[l][j];//向其他点扩展
	}
	printf("%d",Ans);
	fclose(stdin);
	fclose(stdout);
	return 0; 
}