并查集和最小生成树（Prim+Kruskal）

一、并查集
1.初始化

void init(){
	for(int i=1;i<=n;i++)
		father[i]=i;
}

2.查找元素所属集合

int find(int x){ //非递归写法
	while(father[x]!=x)
		x=father[x];
	return x;
}

int find(int x){ //递归写法
	if(father[x]!=x)
		return find(father[x]);
	else
		return x;
}

3.合并集合

void union1(int x,int y){
	x=find(x);
	y=find(y);
	father[y]=x;
}

4.优化（路径压缩） **

int find(int x){  
	if(father[x]!=x)
		father[x]=find(father[x]);
	return father[x];
}

班上有 N 名学生。其中有些人是朋友，有些则不是。他们的友谊具有是传递性。如果已知 A 是 B 的朋友，B 是 C 的朋友，那么我们可以认为 A 也是 C 的朋友。所谓的朋友圈，是指所有朋友的集合。

给定一个 N * N 的矩阵 M，表示班级中学生之间的朋友关系。如果M[i][j] = 1，表示已知第 i 个和 j 个学生互为朋友关系，否则为不知道。你必须输出所有学生中的已知的朋友圈总数。

class Solution {
private:
    vector<int> father;
    int find(int x){
        if(father[x]!=x)
            father[x]=find(father[x]);
        return father[x];
    }
    void union1(int x,int y){
        int x=find(x);
        int y=find(y);
        father[y]=x;
    }
public:
    int findCircleNum(vector<vector<int>>& M) {
        int n=M.size();
        father=vector<int>(n,0);
        for(int i=0;i<n;i++)
            father[i]=i;
        for(int i=0;i<n;i++)
            for(int j=i+1;j<n;j++){
                if(M[i][j])
                    union1(i,j);
            }
        int res=0;
        for(int i=0;i<n;i++){
            if(father[i]==i)
                res++;
        }
        return res;
    }
};

二、最小生成树
1.Prim算法

void MST_Prim(Gragh G){
	int min_weight[G.vexnum];
	int adjvex[G.vexnum];
	for(int i=0;i<G.vexnum;i++){
		min_weight[i]=G.Edge[0][i];
		adjvex[i]=0;
	}  
	int min_arc;
	int min_vex;
	for(int i=1;i<G.vexnum;i++){
		min_arc=MAX;
		for(int j=1;j<G.vexnum;j++){
			if(min_weight[j]!=0&&min_weight[j]<min_arc){
				min_arc=min_weight[j];
				min_vex=j;
			}
		}
		min_weight[min_vex]=0;
		for(int j=0;j<G.vexnum;j++){
			if(min_weight[j]!=0&&G.Edge[min_vex][j]<min_weight[j]){
				min_weight[j]=G.Edge[min_vex][j];
				adjvex[j]=min_vex;
			}
		}
	}
}

2.Kruskal算法

typedef struct
{
    int begin;
    int end;
    int weight;
} Edge;

/* 查找连线顶点的尾部下标 */
int Find(int *father, int f)
{
    while (father[f]!=f)
        f = father[f];

    return f;
}
/* 生成最小生成树 */
void MST_Kruskal(MGraph G)
{
    int i,j,n,m;
    int k=0;
    int parent[MAXVEX];/* 定义一数组用来判断边与边是否形成环路 */

    Edge edges[MAXEDGE];/* 定义边集数组,edge的结构为begin,end,weight,均为整型 */
    /* 此处省略将邻接矩阵G转换为边集数组edges并按权由小到大排列的代码*/
    for (i=0;i<G.numVertexes;i++)
        father[i]=i;
    cout <<"打印最小生成树"<<endl;
    for (i= 0;i<G.numEdges;i++)/* 循环每一条边 */
    {
        n=Find(parent, edges[i].begin);
        m=Find(parent, edges[i].end);
        if(n!=m)/* 假如n与m不等，说明此边没有与现有的生成树形成环路 */
        {
            parent[n]=m;/* 将此边的结尾顶点放入下标为起点的parent中。 */
            /* 表示此顶点已经在生成树集合中 */
            cout<<"("<<edges[i].begin<<", "<<edges[i].end<<") "
                 <<edges[i].weight<<endl;
        }
    }
}

3.例题ccf 数据中心

#include<iostream>
#include<algorithm>
using namespace std;

typedef struct{
	int edge;
	int startPoint;
	int endPoint;
}Edges;

int cmp(Edges a,Edges b){
	return a.edge<b.edge;
}

int find(int *father, int f)
{
    if(father[f]!=f)
        father[f]=find(father,father[f]);
	return father[f];
}

int main(){
	int n,m,root,x,y;
	int Tmax=0;
	int T;
	cin>>n>>m>>root;
	int start,end,arc;
	int father[n];
	for(int i=0;i<n;i++)
		father[i]=i;
	Edges edges[m];
	for(int i=0;i<m;i++){
		cin>>start>>end>>arc;
		edges[i].edge=arc;
		edges[i].startPoint=start;
		edges[i].endPoint=end;
	} 
	sort(edges,edges+m,cmp);
	for(int i=0;i<m;i++){
		x=find(father,edges[i].startPoint);
		y=find(father,edges[i].endPoint);
		if(x!=y){
			father[x]=y;
			T=edges[i].edge;
			if(T>Tmax)
				Tmax=T;
		}
	}
	cout<<Tmax;
	return 0;	
}