HDU 2018计算机复试笔试

第三题

瓜农王大爷去年种西瓜赚了不少钱。看到收入不错，今年他又重新开辟了n个西瓜地。
为了能给他的n个西瓜地顺利的浇上水，对于每个西瓜地他可以选择在本地打井，也可以修管道从另一个瓜地（这个瓜地可能打了井；也可能没打井，他的水也是从其他瓜地引来的）将水引过来。
当然打井和修管道的费用有差别。已知在第i个西瓜地打井需要耗费wi元，在第i、j个西瓜地之间修管道需要耗费pi,j元。
现在的问题是：王大爷要想使所有瓜地都被浇上水，至少需要花费多少钱（打井与修管道的费用和）？
由于瓜地较多，王大爷无法选择在哪些（个）瓜地打井，哪些西瓜地之间修管道。
请你编程帮王大爷做出决策，求出最小费用。

输入格式
第1行，一个正整数n，代表西瓜地的数量。
以下n行，依次给出整数w1…wn（每块西瓜地的打井费用）。
紧接着是一个n*n的整数矩阵，矩阵的第i行第j列的数代表pi,j（两块西瓜地之间建立管道的费用）。每行的两个数之间有一个空格隔开。

样例说明
在第4个瓜地打井（费用为3），然后将第2,3，4个瓜地与第1个瓜地间修管道（费用分别是2,2,2），这样水可以经过管道从4流向1，然后经1再流向2和3；
在第5个瓜地打井（费用为1），5和6之间修管道（费用为9）。
这样一共打了2口井，修了4条管道，能给所有的6个瓜地浇水，费用是：3+2+2+2+1+9=19。

数据规模和约定
对于所有数据，1<=N<=300，1<=wi<=100000；pi,i=0，1<=pi,j=pj,i<=100000。

算法思路
最小生成树 kruskal模板

#include<stdio.h>
#include<algorithm>
using namespace std;
int n,m;
int w[305];
int p[305][305];
int Tree[305];
struct Edge
{
	int from,to,cost;
	bool operator<(const Edge &A) const
	{
		return cost<A.cost; 
	}
	
}edge[100000];


int findRoot(int x)
{
	if(Tree[x]==-1)
		return x;
	else
		return findRoot(Tree[x]);
}


int main()
{
	int n,sum,cnt,i,j,a,b;
	while(scanf("%d",&n),n)
	{
		for(i=0;i<=n;i++)
		{
			Tree[i]=-1;
		}
		m=1;
		for(i=1;i<=n;i++)
		{
			scanf("%d",&w[i]);
			edge[m].from=0;
			edge[m].to=i;
			edge[m].cost=w[i];
			m++;
		}

		for(i=1;i<=n;i++)
		{
			for(j=1;j<=n;j++)
			{
				scanf("%d",&p[i][j]);
			
			if(i<j)
			{
				edge[m].from=i;
				edge[m].to=j;
				edge[m].cost=p[i][j];
				m++;
			}
			}
			

		}
		m--;
		sort(edge+1,edge+1+m);
		sum=0;
		cnt=0;
		for(i=1;i<=m;i++)
		{
			a=findRoot(edge[i].from);
			b=findRoot(edge[i].to);
			if(a!=b)
			{
				Tree[a]=b;
				sum+=edge[i].cost;
				cnt++;
			}
			if(cnt==n)
				break;
		}
		printf("%d\n",sum);

	}
	return 0;
}