并查集模板题 蓝桥幼儿园 java描述

目录

题目描述

输入描述

输入输出样例

完整代码

---

题目描述

蓝桥幼儿园的学生是如此的天真无邪，以至于对他们来说，朋友的朋友就是自己的朋友。

小明是蓝桥幼儿园的老师，这天他决定为学生们举办一个交友活动，活动规则如下：

小明会用红绳连接两名学生，被连中的两个学生将成为朋友。

小明想让所有学生都互相成为朋友，但是蓝桥幼儿园的学生实在太多了，他无法用肉眼判断某两个学生是否为朋友。于是他起来了作为编程大师的你，请你帮忙写程序判断某两个学生是否为朋友（默认自己和自己也是朋友）。

输入描述

第 11 行包含两个正整数 N,MN,M，其中 NN 表示蓝桥幼儿园的学生数量，学生的编号分别为 1\sim N1∼N。

之后的第 2 \sim M+12∼M+1 行每行输入三个整数，op , x , yop,x,y：

• 如果 op = 1op=1，表示小明用红绳连接了学生 xx 和学生 yy 。
• 如果 op=2op=2，请你回答小明学生 xx 和 学生 yy 是否为朋友。

1\leq N,M \leq 2\times10^51≤N,M≤2×105，1 \leq x,y\leq N1≤x,y≤N。

输出描述

对于每个 op=2op=2 的输入，如果 xx 和 yy 是朋友，则输出一行 YES，否则输出一行 NO。

输入输出样例

示例 1

输入

5 5 
2 1 2
1 1 3
2 1 3
1 2 3 
2 1 2

输出

NO
YES
YES

思路：并查集模板题，把学生连接起来，如果根节点 find(x)==find(y) 则是朋友

查找根节点（路径压缩）

public static int find(int x)
	{
		if(x == fa[x]) return x;
		else
		{
			fa[x] = find(fa[x]);
			return fa[x];
		}
	}

将学生连接起来

public static void union(int x, int y)
	{
		fa[find(x)] = find(y);		
	}

完整代码

import java.util.*;
import java.io.*;

public class Main {
	static int[] fa;
	public static void main(String[] args) throws IOException{
		Scanner scan = new Scanner(System.in);
		int N=scan.nextInt();
		fa = new int[N+1];
		for(int i=1;i<=N;i++) fa[i] = i;
		
		int M=scan.nextInt();
		for(int i=0;i<M;i++)
		{
			int op = scan.nextInt();
			int x = scan.nextInt();
			int y = scan.nextInt();
			if(op==1)
			{
				union(x,y);
			}
			else if(op==2)
			{
				if(find(x)==find(y)) System.out.println("YES");
				else System.out.println("NO");
			}
			
		}
		
		
	}
	
	public static void union(int x, int y)
	{
		fa[find(x)] = find(y);		
	}
	public static int find(int x)
	{
		if(x == fa[x]) return x;
		else
		{
			fa[x] = find(fa[x]);
			return fa[x];
		}
	}
}