53-蓝桥幼儿园(蓝桥杯)
迪丽瓦拉
2025-05-31 17:53:53
0

题目描述

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

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

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

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

输入描述

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

之后的第 2∼M+1 行每行输入三个整数,op,x,y:

  • 如果 op=1,表示小明用红绳连接了学生 x 和学生 y 。

  • 如果 op=2,请你回答小明学生 x 和 学生 y 是否为朋友。

1≤N,M≤2×105,1≤x,y≤N。

输出描述

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

输入输出样例

示例 1

输入
5 5 
2 1 2
1 1 3
2 1 3
1 2 3 
2 1 2
输出
NO
YES
YES

运行限制

  • 最大运行时间:5s

  • 最大运行内存: 256M

源码:

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;public class 蓝桥幼儿园 {static class UnionFind{int[] parent;int[] rank;public UnionFind(int n) {parent=new int[n];for (int i = 0; i < n; i++) {parent[i]=i;}rank=new int[n];Arrays.fill(rank, 1);}public int find(int ind) {if(parent[ind]!=ind) {parent[ind]=find(parent[ind]);}return parent[ind];}public void unite(int ind1,int ind2) {int root1=find(ind1),root2=find(ind2);if (root1!=root2) {if (rank[root1]

相关内容