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

Description–

FY决定给N(1<=N<=300)个草地给水，草地编号为1到N。他可以在该草地处建立一口井，也可以建立管道从别的草地引水进来，在第i块草地处挖一口井需要花费Wi(1<=Wi<=100,000)，连接井i和井j的管道需要花费Pij(1<=Pij<=100,000,Pij=Pji;Pii=0)。
计算最少需要花费多少才能保证每个草地都有水。

---

Input–

第一行，一个整数N
第2到N+1行，第i+1行输入Wi
第N+2到2N+1行，每行N个空格隔开的整数，表示Pij

Output–

输出最少花费。

---

Sample Input–

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

Sample Output–

9

---

说明–

【样例解释】
FJ可以在草地4建井，并把草地2，3，4与草地1之间建立管道，花费为3+2+2+2=9

---

解题思路–

新建一个0号节点，把它连到每个点的花费，设为新开凿的花费，再用0号节点做最小生成树。

也可以去看看lyf的骗分方法

---

代码–

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int n,mn=1,ans,a[305],dis[305],fy[305][305];
bool pd[305];
int main()
{
	freopen("water.in","r",stdin);
	freopen("water.out","w",stdout);
	
	scanf("%d",&n);
	memset(dis,0x7f,sizeof(dis));//初始化一个较大的数
	for (int i=1;i<=n;++i)
	{
		scanf("%d",&a[i]);
		dis[i]=a[i];
	}
	for (int i=1;i<=n;++i)
	  for (int j=1;j<=n;++j)
	    scanf("%d",&fy[i][j]);
    for (int i=1;i<=n;++i)
    {
    	int fff=0;
    	for (int j=1;j<=n;++j)
    	  if (!pd[j] && (dis[j]<dis[fff]))
    	    fff=j;//取出现有路径的最小花费路径
    	pd[fff]=1;
    	for (int j=1;j<=n;++j)
    	  if (!pd[j] && (fy[fff][j]<dis[j]))
            dis[j]=fy[fff][j];//更新连到j点的最小花费路径
	}
	for (int i=1;i<=n;++i)
	  if (pd[i])
	    ans+=dis[i];//累积答案
	printf("%d\n",ans);
	
	return 0;
}