week-6-C - 掌握魔法の东东 I(最小生成树问题）

题意

东东在老家农村无聊，想种田。农田有 n 块，编号从 1~n。种田要灌氵
众所周知东东是一个魔法师，他可以消耗一定的 MP 在一块田上施展魔法，使得黄河之水天上来。他也可以消耗一定的 MP 在两块田的渠上建立传送门，使得这块田引用那块有水的田的水。 (1<=n<=3e2)
黄河之水天上来的消耗是 Wi，i 是农田编号 (1<=Wi<=1e5)
建立传送门的消耗是 Pij，i、j 是农田编号 (1<= Pij <=1e5, Pij = Pji, Pii =0)
东东为所有的田灌氵的最小消耗

输入输出说明
intput :
第1行：一个数n
第2行到第n+1行：数wi
第n+2行到第2n+1行：矩阵即pij矩阵
output:
东东最小消耗的MP值

样例：
input:
4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0
output:
9

思路

• 这个题是为了求最小的消耗，而且每一块农田都与其他农田之间存在可选择的消耗，而且出现了取黄河水，那么我们可以把黄河当作另一块农田，然后这就是一个图求最短路径的问题，我们就可以使用最小生成树；
• 我们使用一个edge边集，将这些农田之间的路径记录下来，edge包括路劲长度、起始点和终止点；
• 然后进行edge排序，每次取出最小的那个边，然后使用并查集进行合并，合并n次以后得到的min_path就是我们想得到的最少消耗；

总结

此题中构建的图是一个连通图，不连通的话，还需要进行判断图是否连通；
注：农田之间的路径是双向的，但黄河水那个不是双向的；

代码

#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
struct num
{
	int v1;//标号
	int v2;
	int edge;//消耗
	bool operator<(const num& p)
	{
		return edge < p.edge;
	}
	num& operator=(const num& p)
	{
		v1 = p.v1;
		v2 = p.v2;
		edge = p.edge;
		return *this;
	}
};
int a[1000];
int n;
void init()//并查集的初始化
{
	for (int i = 1; i < n+3; i++)
		a[i] = i;
}
int find(int x)//并查集的查询
{
	return a[x];
}
void unite(int x, int y)
{
	for (int i = 1; i < n+3; i++)
	{
		if (a[i] == y)
			a[i] = x;
	}

}

long long sum = 0;
int count = 0;
void min_way(struct num* edge)
{
	int leng = n * n+n;
	sort(edge, edge + leng);
	init();
	int c = 0;
	for (int i = 0; i < leng; i++)
	{
		struct num temp = edge[i];
		int x=find(edge[i].v1);
		int y = find(edge[i].v2);
		if (x != y)
		{
			unite(x, y);
			sum = sum + edge[i].edge;
			c++;
		}
		if (c == n+1)
			break;
	}
}
int main()
{
	cin >> n;
	struct num* p = new struct num[n + 1];
	for (int i = 1; i <= n; i++) {
		cin >> p[i].edge;   //直接灌溉
		p[i].v1 = i;
		p[i].v2 = i;
	}
	int leng = n + 1;
	//sort(p, p + leng);//第一步必须从黄河水来
	struct num **q= new struct num*[n + 2];//建立传送门
	struct num* edge = new struct num[n * n + n];
	int cou = 0;
	for (int i = 1; i <= n; i++)
		q[i] = new struct num[n + 2];
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= n; j++)
		{
			if (i == j)
			{
				cin >> q[i][j].edge;
				q[i][j].edge = 100000;
				q[i][j].v1 = i;
				q[i][j].v2 = j;
				edge[cou] = q[i][j];
				cou++;
			}
			else {
				cin >> q[i][j].edge;
				q[i][j].v1 = i;
				q[i][j].v2 = j;
				edge[cou] = q[i][j];
				cou++;
			}
		}
	for (int i = 1; i <= n; i++)
	{
		q[i][n + 1].edge = p[i].edge;
		q[i][n + 1].v1 = i;
		q[i][n + 1].v2 = n + 1;
		edge[cou] = q[i][n+1];
		cou++;
	}
	min_way(edge);
	cout << sum << endl;
 	return 0;
}