忽略题目背景,要计算的是一个数根节点的子树中哪个子树的节点个数最多。
输入格式是先输入n,接下来有n-1行,每行为a,b,表示a和b之间有边。用并查集的思路可解。 先上源码。
import java.util.Scanner;
/**
* @author JYMiracle
* @date 2019年4月13日
* @time 下午8:51:44
*/
public class Main {
static int[] v = new int[100010];
static int[] sum = new int[100010];
static int getroot(int a)
{
while(a!=v[a])
a=v[a];
return a;
}
public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner in = new Scanner(System.in);
int n = in.nextInt();
for(int i=0;i<=n;i++){
v[i]=i; //初始化每个点都连在自己身上
}
for(int i=0;i<n-1;i++)
{
int a,b; //输入的a b 表示a和b有边
a= in.nextInt();
b = in.nextInt();
int aa,bb;
aa=getroot(a);
bb=getroot(b);
// aa = a;
// bb = b;
if(aa!=bb) //如果他们目前不是一个集合
{
v[a]=b; //把b连到a的阵营里面,a到b有边
}
}
// for (int i = 0; i < n; i++) {
// System.out.println(v[i]);
// }
for(int i=2;i<=n;i++)
{
int tem=i;
while(tem!=v[tem]){
sum[tem]++;
tem=v[tem];
}
}
int maxn=sum[1];
for(int i=1;i<=n;i++)
{
//System.out.println(sum[i]);
if(sum[i]>maxn)
{
maxn=sum[i];
}
}
System.out.println(maxn);
}
}
总结并查集的思路:用v[a] = b 表示a到b有边,初始化v[i] = i;重要的一个步骤就是寻找每个集合的头
int getroot(int a){
while(a!=v[a]) //a连的不是自己
a=v[a]; //a=v[a]是向上搜寻头,直到a==v[a]说明找到了这个集合的头
return a;
}
当输入新的两个有边的节点时,先判断它们的集合是否相同(即向上搜寻头,判断头是否相同),若不同,即把后者加到前者的集合中(v[a]=b)。在处理每个集合的节点个数时,一种做法是把集合内除了头之外的节点都连到头,这样看有多少v[i] = root;在这里对于每个v[i]向上搜寻头的同时,维护一个sum[i]表示以i为头的下面的节点有多少,最后选最大者。
这是一篇讲解并查集入门非常生动清晰的博客https://www.cnblogs.com/xzxl/p/7226557.html,博主一定是个有趣的灵魂。