并查集算法练习-判断亲戚关系

题目描述 或许你并不知道，你的某个朋友是你的亲戚。他可能是你的曾祖父的外公的女婿的外甥女的表姐的孙子。如果能得到完整的家谱，判断两个人是否亲戚应该是可行的，但如果两个人的最近公共祖先与他们相隔好几代，使得家谱十分庞大，那么检验亲戚关系实非人力所能及。在这种情况下，最好的帮手就是计算机。为了将问题简化，你将得到一些亲戚关系的信息，如Marry和Tom是亲戚，Tom和Ben是亲戚，等等。从这些信息中，你可以推出Marry和Ben是亲戚。请写一个程序，对于我们的关于亲戚关系的提问，以最快的速度给出答案。

输入 输入由两部分组成。 第一部分以N，M开始。N为问题涉及的人的个数(1≤N≤20000)。这些人的编号为1,2,3,…, N。下面有M行(1≤M≤1 000 000)，每行有两个数ai, bi，表示已知ai和bi是亲戚。 第二部分以Q开始。以下Q行有Q个询问(1≤Q≤1 000 000)，每行为ci, di，表示询问ci和di是否为亲戚。

输出 对于每个询问ci, di，输出一行：若ci和di为亲戚，则输出“Yes”，否则输出“No”。

样例输入 10 7 2 4 5 7 1 3 8 9 1 2 5 6 2 3 3 3 4 7 10 8 9

样例输出 Yes No Yes

import java.util.Scanner;

public class Solution {
 public Solution(){

 }
 public void union(int[][] relation,int[] GroupSign){//合并集合
     for(int i=0,j;i<relation.length;i++){
         int sign=GroupSign[relation[i][1]-1];
         for(j=0;j<GroupSign.length;j++){
             if(GroupSign[j]==sign){
                 GroupSign[j]=GroupSign[relation[i][0]-1];//把另一个集合的所有元素加入另外一个集合中，更改所有元素的代表元
             }
         }
     }
 }
 public boolean find(int[] GroupSign,int signA,int signB){//判断两个元素是否属于一个集合，如果代表元相同则表示属于一个集合
     return GroupSign[signA-1]==GroupSign[signB-1];
 }
    public static void main(String[] args) {
        Solution solution=new Solution();
     int n,m,p,GroupCount=0;
        Scanner reader=new Scanner(System.in);
        System.out.println("请输入人数:");
        n=reader.nextInt();
        System.out.println("请输入有几种亲戚关系");
        m=reader.nextInt();
        System.out.println("请输入想要询问的亲戚关系的对数:");
        p=reader.nextInt();
        int[][]relation=new int[m][2];//关系数组，标识有多少组关系
        int[]GroupSign=new int[n];//初始化每个元素自身为一个集合
        GroupCount=n;
        for (int j=0;j<n;j++){
            GroupSign[j]=j;//初始化代表元为自身
        }
        for (int i=0;i<m;i++){
            System.out.println("请输入一对关系以空格间隔");
            relation[i][0]=reader.nextInt();
            relation[i][1]=reader.nextInt();
        }
        solution.union(relation,GroupSign);
        int x,y;
        for(int k=0;k<p;k++){
            System.out.println("请输入一对试验组合:");
            x=reader.nextInt();
            y=reader.nextInt();
            if(solution.find(GroupSign,x,y))
                System.out.println("Yes");
            else
                System.out.println("No");
        }
    }

}