蓝桥杯-蓝桥幼儿园-python

目录

题目描述

输入描述

输出描述

输入输出样例

运行限制

思路： 

代码

---

题目描述

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

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

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

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

输入描述

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

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

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

1≤N,M≤2*10^5 ,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

运行限制

• 最大运行时间：3s
• 最大运行内存: 256M

思路： 

采用并查集的方法。

1。设计初始化函数init()，初始化数组s，将编号为1~N的N个对象划分为不相交的集合，在每个集合中选择一个元素代表整个集合。

2.设计查找函数find(),通过递归的方法返回x的所在集合的代表元素。

3.设置合并函数merge()，寻找x,y的根节点，如果根节点不一致，则改成一致

主函数：

当op==1时，读入x,y合并集合s[x],s[y]，使得s[x]=s[y]

当op==2时，寻找x和y的根节点，通过比较根节点是否一致判断x和y是否在一个集合中。如果在，说明x,y是朋友，输出YES，否则输出NO。

代码

N=800000
s=[]

def init():
    for i in range(N):
        s.append(i)

def find(a):
    if a!=s[a]:
        s[a]=find(s[a])
    return s[a]

def merge(x,y):
    x=find(x)
    y=find(y)
    if x!=y:
        s[x]=s[y]

n,m=map(int,input().split())
init()
for i in range(m):
    op,x,y=map(int,input().split())
    if op==1 :
        merge(x,y)
    if op==2 :
        if find(x)==find(y):
            print("YES")
        else:
            print("NO")