初学最小生成树 （B - 新的开始）

今天是寒假集训回家后的第一天，在学长的带领下学习了最小生成树。借此来加深一下对这个算法的认识和理解。

首先我们要明白是什么是最小生成树：

简单来说，最小生成树就是我们在一个图里，把n个点通过n-1条边连接，并且所权数和最小；

最小生成树比较常用的是kruskal 和 prim 算法。

kruskal

这个算法和并查集紧密联系。比较类似于求带权并查集的最小值。

首先我们知道，最小生成树是为了找一个耗费最少的方法把全部点连接起来。而并查集也有连接的功能，我们把带权并查集的权数看作是花费，那么我们通过排序之后，从小到大的顺序，每次连接都把权数加起来，那么最早满足全部连接的权数和一定是最小的。也就是满足了最小生成树的目的。

prim

这个算法就是类似于扩散，采用邻接矩阵储存边。从某一个点出发，以最优的方式连接一个图。

每次循环都会是一次遍历，选取最最短的路线且没被选取过的进行连接。

以下边一道题为例：

通过对题意分析，我们知道，这道题的题意就是说以最小的花费，把所有矿井连接在一起。那么就是我们所说的最小生成树。

样例分析

首先输入的是矿井个数，接着是自己建造的费用。接下来的矩阵，横着看，第i个矿井和其他的矿井所需要费用，比如说第一行是0 2 2 2，是1号矿井分别和1 2 3 4号相连的费用。下边这个图就是对样例结果的分析。

第一种解法（kruskal）

通过结构体把两个矿井和费用进行连接，进行排序，通过并查集的思想进行相连。

Code：

#include<stdio.h>
#include<algorithm>
#include<iostream>
#include<set>
#include<map>
#include<vector>
#include<string.h>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
int xa[1000010];
int ans = 0;
struct ch 
{
	int x, y, z;
}s[1000010]; 
int find(int x) // 找祖宗节点 
{
	if(xa[x] == x)
	{
		return x;
	}
	else 
	{
		return  xa[x] = find(xa[x]);
	}
}
void fa(int x, int y, int z) // 连接矿井 
{
	int a = find(x);
	int b = find(y);
	if(a == b)
	{
		return;
	}
	else 
	{
		xa[b] = a; 
		ans += z;
	}
}
bool cmp(ch a, ch b)
{
	return a.z <b.z;
}
int main()
{
	ios::sync_with_stdio(false);
	ll n , hf;
	cin >> n ;
	for(int i = 1; i <= n; i ++)
	{
		xa[i] = i;
	}
	int num = 1;
	for(int i = 1; i <= n; i ++)
	{
		cin >> hf;
		s[num ].x = i;
		s[num ].y = 0;//把0记为空，即对自己连接的费用。 
		s[num ++].z = hf;
	}	
	for(int i = 1; i <= n; i ++)
	{
		for(int j = 1; j <= n; j ++)
		{
			cin >> hf;
			s[num ].x = i; //i，j分别代表矿井号，即i j矿井相连所需费用为hf. 
			s[num ].y = j;
			s[num ++].z = hf; 
		}
	}
	int xx = n * n + n; //由于我们把所有情况都存在一个结构体里面，所有长度要远超于n； 
	sort(s + 1,s + 1 + xx,cmp);
	for(int i = 1; i <= xx ; i ++)
	{
		fa(s[i].x, s[i].y, s[i].z); 
	}
	cout << ans << endl;
}

 第二种解法（prim）

prim的原理是以一个点为初始点，从它开始往四周扩散。选取花费最小的方式连接，最后通过n-1条路将所有连接起来。这题来说选取一个矿井，然后通过它连接其他几个，构成最小生成树。

由图表可知，我们第一次i循环的时候是选取一个未命名的点，然后对自身连接，再和其他的相连，所以第一次ans的值为0.（即选点）。第二次i循环便是对自身连接并对点命名。

Code：

#include<stdio.h>
#include<algorithm>
#include<iostream>
#include<set>
#include<map>
#include<vector>
#include<queue>
#include<string.h>
#include<cmath>
using namespace std;
typedef long long ll;
int a[1010][1010]; //存图 
int dist[10010];//记录更新后的距离 
bool st[10010];//状态 
int n;
int prim()
{
	int ans = 0;
	dist[0] = 0;//所选取第一个点 距离为0；第二次i循环会加上自己建造花费最小的值。 
	for(int i = 0; i <= n; i++)
	{
		int t = -1;// 此处是为了保证下下边循环第一次可以顺利进入，防止re； 
		for(int j = 0; j <= n; j++)
		{
			if(!st[j] && (t == -1 || dist[t] > dist[j]))//找没被选取且最近的一个。 
			t = j;
		}
		ans += dist[t]; // 总费用。
	//	cout << ans << endl; 
		st[t] = 1;  // 选取状态； 
		for(int j = 0; j <= n; j++)
		{
			dist[j] = min(dist[j], a[t][j]); //以t为中心，更新到其他点的值。 
		//	cout << dist[j] << " ";
		}
		//cout << endl;
	}
	return ans;
}
int main()
{
	ios::sync_with_stdio(false);
	cin >> n;
	memset(dist,0x3f3f3f3f,sizeof(dist));//初始化使各边不相邻。 
	memset(a,0x3f3f3f3f,sizeof(a));
	for(int i = 1; i <= n; i++) //用第0条边记录自己建造的费用。 
	{
		cin >> a[0][i];
		a[i][0] = a[0][i];
	}
	for(int i = 1; i <= n; i++)
	{
		for(int j = 1; j <= n; j++)
		{
			cin >> a[i][j];  
		}
	}
	int sum = prim();
	cout << sum << endl;
} 

愿各位共同进步。