Tree - NHIP第三题解题报告 - 并查集入门练习

#include <cstdio>
using namespace std;
struct node {
int link;
int point_num;
int line_num;
bool haveFind;
};
node point[1000001];
int find_root(int t) {
int root,lt;
root=t-1;
//查找
while(point[root].link!=root) {
root=point[root].link;
}
//并查集的路径优化
while(point[t-1].link!=root) {
lt=point[t-1].link; //存储当前点的父亲
point[t-1].link=root; //将当前点设置为直接指向Root
t=lt+1;   //将当前点的父亲赋回给循环变量
}
return root;
}
inline void union_set(int f,int s) {
point[ find_root( s ) ].point_num+=point[ find_root( f ) ].point_num;
point[ find_root( s ) ].line_num+=point[ find_root( f ) ].line_num+1;
point[ find_root( f ) ].link = find_root( s );
}
int main() {
int n;
int m;
scanf("%d %d",&n,&m);
//建立整个并查集的结构
for (int i=0; i<n; ++i) {
point[i].link=i;
point[i].point_num=1;
point[i].line_num=0;
point[i].haveFind=false;
}
int  f,s;

//建立并查集的关系
for (int i=0; i<m; ++i) {
scanf("%d %d",&f,&s);
//合并两个图
if (find_root(f)!=find_root(s)) {
union_set(f,s);
}
}
long long int ans = 0;
for (int i=0; i<n; ++i) {
printf("%d",find_root(i));
if ((point[find_root(i)].point_num-1 == point[find_root(i)].line_num) && !point[find_root(i)].haveFind) {	//是否是树，是否访问过
point[find_root(i)].haveFind=true;	//设为访问过
++ans;	//累计答案
}
}
printf("%d",ans);	//输出答案
return 0;
}