[jzoj]2015. 给水

题目描述

FJ决定给N(1<=N<=300)个草地给水，草地编号为1到N。他可以在该草地处建立一口井，也可以建立管道从别的草地引水进来，在第i块草地处挖一口井需要花费W_i(1<=W_i<=100,000)，连接井i和井j的管道需要花费P_ij(1<=P_ij<=100,000,P_ij=P_ji;P_ii=0)。

计算最少需要花费多少才能保证每个草地都有水。

输入

第一行，一个整数N
第2到N+1行，第$i+1$行输入$W_i$
第$N+2$到$2N+1$行，每行$N$个空格隔开的整数，表示$P_{i,j}$

输出

输出最少花费。

题解

最小生成树

Code

#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);
	cin >> n;
	memset(dis,0x7f,sizeof(dis));
	for (int i=1;i<=n;++i)
	{
		cin>>a[i];
		dis[i]=a[i];
	}
	for (int i=1;i<=n;++i)
	  for (int j=1;j<=n;++j)
	    cin>>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];
	}
	for (int i=1;i<=n;++i)
	  if (pd[i])
	    ans+=dis[i];
	cout<<ans;
	
	return 0;
}