目录
题目描述
蓝桥幼儿园的学生是如此的天真无邪,以至于对他们来说,朋友的朋友就是自己的朋友。
小明是蓝桥幼儿园的老师,这天他决定为学生们举办一个交友活动,活动规则如下:
小明会用红绳连接两名学生,被连中的两个学生将成为朋友。
小明想让所有学生都互相成为朋友,但是蓝桥幼儿园的学生实在太多了,他无法用肉眼判断某两个学生是否为朋友。于是他起来了作为编程大师的你,请你帮忙写程序判断某两个学生是否为朋友(默认自己和自己也是朋友)。
输入描述
第 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];
}
}
}