HDU：杭电 2018 复试真题汇总

1. 题目描述

瓜农王大爷去年种西瓜赚了不少钱。看到收入不错，今年他又重新开辟了 $n$ 个西瓜地。

为了能给他的 $n$ 个西瓜地顺利的浇上水，对于每个西瓜地他可以选择在本地打井，也可以修管道从另一个瓜地（这个瓜地可能打了井；也可能没打井，它的水也是从其它瓜地引来的）将水引过来。

当然打井和修管道的费用有差别。已知在第 $i$ 个西瓜地打井需要耗费一定费用，在第 $i$ 、 $j$ 个西瓜地之间修管道需要耗费一定费用。现在的问题是：王大爷要想使所有瓜地都被浇上水，至少需要花费多少钱（打井与修管道的费用和）？

由于瓜地较多，王大爷无法选择在哪些（个）瓜地打井，哪些西瓜地之间修管道。

请你编程帮王大爷做出决策求出最小费用。

输入格式

第 1 行，一个正整数 $n$ ，代表西瓜地的数量。

第 2 行，依次给出整数 $w_1, w_2 \\dots w_n$ （每块西瓜地的打井费用），两个数之间有一个空格隔开。

紧接着是一个 $n \\times n$ 的整数矩阵，矩阵的第 $i$ 行第 $j$ 列的数代表（两块西瓜地之间建立管道的费用），每行的两个数之间有一个空格隔开。

输出格式

一个正整数，所求的最小花费。

数据规模和约定

对于所有数据， $1 \\le n \\le 300$ ，打井费用 $1 \\le w_i \\le 100000$ ，管道费用及矩阵对角线元素（通常约定为0或无意义）， $1 \\le p_{i,j} \\le 100000$ 。（注：原图此处印刷模糊，此为根据常规题意推断的合理范围）

	样例 1	样例 2	样例 3
输入	`6` `5 4 4 3 1 20` `0 2 2 2 9 9...` (矩阵首行)	`6` `10 4 5 3 1 20` `0 2 2 2 9 8...` (矩阵首行)	`1` `1` `0`
输出	`19`	`18`	`1`

cpp 复制代码

#include<iostream>
#include<vector>
#include<string>
#include<stdio.h>
#include<algorithm>

using namespace std;
int melon[305];
int vis[305];
int parent[305];

typedef struct {
	int begin;
	int end;
	int value;
}edge;

bool cmp(edge A,edge B) {
	return A.value<B.value;
}

int findroot(int x){
	if(parent[x]==x) return x;
	return parent[x]=findroot(parent[x]);
}

int main() {
	int n;
	while (cin >> n) {
		vector<edge>s1;
		edge temp;
		
		for(int i=0;i<n;i++) {
			cin>>melon[i];
			temp.begin=n;
			temp.end=i;
			temp.value=melon[i];
			s1.push_back(temp);	
		}
		
		for(int i=0;i<n;i++) {
			for(int j=0;j<n;j++) {
				temp.begin=i;
				temp.end=j;
				cin>>temp.value;
				s1.push_back(temp);
			}
		}
		
		sort(s1.begin(),s1.end(),cmp);
		
		int sum=0;
		int count=0;
		
		for (int i = 0; i <= n; i++) {
			parent[i] = i; 
		}
		
		for(int i=0;i<s1.size();i++) {
			temp=s1[i];
			if(findroot(temp.begin)!=findroot(temp.end)) {
				parent[findroot(temp.end)]=findroot(temp.begin);
				sum+=temp.value;
				count++;
				vis[temp.begin]=1;
				vis[temp.end]=1;
			}
			if(count==n)
				break;
		}
		cout<<sum<<endl;
	}
	return 0;
}