并查集实现

  常可用来判断两个元素是否在同一个集合中,n^2的复杂度太大。

 1 #include <stdio.h>
 2 
 3 #define MAX 100
 4 int parent[MAX];
 5 
 6 int find(int x)
 7 {
 8     if(x<0||x>=MAX)
 9     {
10         return -1;
11     }
12     if(parent[x]==x)
13     {
14         return parent[x];
15     }else
16     {
17         return parent[x] = find(parent[x]);
18     }
19 }
20 
21 
22 void init()
23 {
24     int i;
25     for(i=0;i<MAX;i++)
26     {
27         parent[i] = i;
28     }
29 }
30 
31 void my_union(int i, int j)
32 {
33     if(i<0 || i>=MAX || j<0 || j>=MAX)
34     {
35         return ;
36     }
37     int px = find(i);
38         int py = find(j);
39         parent[px] = py;
40 }
41 
42 int main()
43 {
44     int i;
45     init();
46     my_union(0, 1);
47     for(i=0;i<10;i++)
48     {
49         printf("%d\n", find(i));
50     }
51     return 0;
52 }
53 /*
54 秩的压缩,即总是把含孩子少的合并到孩子多的子树上。以减少不平衡树的产生。但是话说这个优化感觉有限。
55 路径压缩。实际是记忆搜索或者伸展树的变种。
56 */


 

并查集是一种数据结构,可以有效地解决团伙问题。团伙问题指的是在一个群体中,找出其中的团伙关系。举个例子,假设有一群人,其中有些人是朋友,有些人是陌生人,现在需要找出所有的朋友团伙。这个问题可以用并查集来解决。 具体实现过程如下: 1. 初始化并查集,将每个人看作一个单独的团伙,即每个人的父节点都是自己。 2. 遍历所有的朋友关系,将每个朋友关系所在的两个人合并到同一个团伙中。合并的过程中,需要找到两个人所在团伙的根节点,将其中一个根节点的父节点指向另一个根节点。 3. 遍历所有的人,统计每个团伙的人数。具体方法是遍历每个人,找到该人所在团伙的根节点,然后统计该根节点下所有子节点的个数。 4. 输出所有团伙的人数。 下面是并查集实现团伙问题的示例代码(假设已经读入了朋友关系列表friend_list): ```python # 初始化并查集 parent = {} for i in range(len(friend_list)): parent[i] = i # 合并朋友关系所在的团伙 for f1, f2 in friend_list: root1 = find_root(parent, f1) root2 = find_root(parent, f2) if root1 != root2: parent[root1] = root2 # 统计每个团伙的人数 group_sizes = {} for i in range(len(friend_list)): root = find_root(parent, i) if root not in group_sizes: group_sizes[root] = 1 else: group_sizes[root] += 1 # 输出每个团伙的人数 for root, size in group_sizes.items(): print("团伙{}有{}个人".format(root, size)) # 找到节点的根节点 def find_root(parent, node): while parent[node] != node: node = parent[node] return node ``` 这个代码实现了一个简单的并查集,并可以用来解决团伙问题。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值