给水

给水

题目大意

要给N个草地给水，草地编号为1到N。可以在该草地处建立一口井，也可以建立管道从别的草地引水进来，在第i块草地处挖一口井需要花费Wi，连接井i和井j的管道需要花费Pij。
求最少需要花费多少才能保证每个草地都有水。

输入样例

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

输出样例

9

样例解释

可以在草地4建井，并把草地2，3，4与草地1之间建立管道，花费为3+2+2+2=9

数据范围

$1<=N<=300$
$1<=Wi<=100,000$
$1<=Pij<=100,000$
$Pij=Pji$
$Pii=0$

思路

这道题而其实就是一道最小生成树，只是要处理打井的费用。
我们可以建一个点0，连接所有的点，长度就为打井的费用。这样，我们从零开始进行kruskal算法，就可以算出答案了。

代码

#include<cstdio>
#include<algorithm> 
using namespace std;
struct note
{
	int x,y,l;
}a[100001];
int n,f[301],k,answer,u;
int find(int x)//查找
{
	if (f[x]==x) return x;
	return f[x]=find(f[x]);
}
void connect(int x,int y)//连通
{
	int xx=find(x),yy=find(y);
	if (xx<yy) f[xx]=yy;
	else f[yy]=xx;
}
bool cmp(note x,note y)
{
	return x.l<y.l;
}
int main()
{
	freopen("water.in","r",stdin);
	freopen("water.out","w",stdout); 
	scanf("%d",&n);//读入
	for (int i=1;i<=n;i++)
	{
		scanf("%d",&a[++k].l);//读入
		a[k]=(note){0,i,a[k].l};//标记
	}
	for (int i=1;i<=n;i++)
	{
		f[i]=i;//初始化
		for (int j=1;j<=n;j++)
		{
			scanf("%d",&a[++k].l);//读入
			if (a[k].l==0) k--;//没有路的
			else a[k]=(note){i,j,a[k].l};//标记
		}
	}
	sort(a+1,a+k+1,cmp);//排序（按长度从小到大排）
	for (int i=1;i<=k;i++)//kruskal算法
	if (find(a[i].x)!=find(a[i].y))
	{
		u++;
		connect(a[i].x,a[i].y);
		answer+=a[i].l;
		if (u==n) break;
	}
	printf("%d",answer);//输出
	fclose(stdin);
	fclose(stdout);
	return 0;
}