STL - 并查集

1、并查集原理

        在一些应用问题中,需要将n个不同的元素划分成一些不相交的集合;开始时,每个元素自成一个 单元素集合,然后按一定的规律将归于同一组元素的集合合并;在此过程中要反复用到查询某一 个元素归属于那个集合的运算,适合于描述这类问题的抽象数据类型称为并查集(union-find set)

        比如:某公司今年校招全国总共招生10人,西安招4人,成都招3人,武汉招3人,10个人来自不 同的学校,起先互不相识,每个学生都是一个独立的小团体,现给这些学生进行编号:{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}; 给以下数组用来存储该小集体,数组中的数字代表:该小集体中具有成员的个 数。(负号下文解释)

        毕业后,学生们要去公司上班,每个地方的学生自发组织成小分队一起上路,于是: 西安学生小分队s1={0,6,7,8},成都学生小分队s2={1,4,9},武汉学生小分队s3={2,3,5}就相互认识 了,10个人形成了三个小团体。假设右三个群主0,1,2担任队长,负责大家的出行

一趟火车之旅后,每个小分队成员就互相熟悉,称为了一个朋友圈

从上图可以看出:编号6,7,8同学属于0号小分队,该小分队中有4人(包含队长0);编号为4和9的同 学属于1号小分队,该小分队有3人(包含队长1),编号为3和5的同学属于2号小分队,该小分队有3 个人(包含队长1)

仔细观察数组中内融化,可以得出以下结论:

  1. 数组的下标对应集合中元素的编号
  2. 数组中如果为负数,负号代表根,数字代表该集合中元素个数
  3. 数组中如果为非负数,代表该元素双亲在数组中的下标

在公司工作一段时间后,西安小分队中8号同学与成都小分队1号同学奇迹般的走到了一起,两个 小圈子的学生相互介绍,最后成为了一个小圈子:

现在0集合有7个人,2集合有3个人,总共两个朋友圈

 通过以上例子可知,并查集一般可以解决一下问题:

     1. 查找元素属于哪个集合

        沿着数组表示树形关系以上一直找到根(即:树中中元素为负数的位置)

     2. 查看两个元素是否属于同一个集合

        沿着数组表示的树形关系往上一直找到树的根,如果根相同表明在同一个集合,否则不在

     3. 将两个集合归并成一个集合

  • 将两个集合中的元素合并
  • 将一个集合名称改成另一个集合的名称

     4. 集合的个数

        遍历数组,数组中元素为负数的个数即为集合的个数

2、并查集实现

class UnionFindSet
{
public:
  // 初始时,将数组中元素全部设置为1
     UnionFindSet(size_t size)
     : _ufs(size, -1)
     {}
  // 给一个元素的编号,找到该元素所在集合的名称
     int FindRoot(int index)
     {
      // 如果数组中存储的是负数,找到,否则一直继续
         while(_ufs[index] >= 0)
         {
             index = _ufs[index];
         }
     
         return index;
     }

     bool Union(int x1, int x2)
     {
         int root1 = FindRoot(x1);
         int root2 = FindRoot(x2);
     
      // x1已经与x2在同一个集合
         if(root1 == root2)
             return false;
     
      // 将两个集合中元素合并
         _ufs[root1] += _ufs[root2];
         
      // 将其中一个集合名称改变成另外一个
         _ufs[root2] = root1;
         return true;
     }
 // 数组中负数的个数,即为集合的个数
    size_t Count()const
    {
        size_t count = 0;
        for(auto e : _ufs)
        {
             if(e < 0)
                 ++count;
         }
     
         return count;
    }
     
private:
    vector<int> _ufs;
};

3、并查集应用

  1. 省份数量
  2. 等式方程的可满足性
  • 16
    点赞
  • 26
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
STL-10数据集是一个包含10个类别的图像数据集,包括飞机、鸟、汽车、猫、鹿、狗、青蛙、马、船和卡车。每个类别包含5000张32x32的彩色图像,其中500张用于训练,100张用于验证,剩余的用于测试。 在本教程中,我们将学习如何加载STL-10数据集。 1.下载数据集 首先,我们需要下载STL-10数据集。可以从以下网站下载:http://ai.stanford.edu/~acoates/stl10/ 2.解压缩数据集 下载完成后,将文件解压缩到指定的目录中。 3.加载数据集 我们可以使用Python中的NumPy和pickle模块来加载STL-10数据集。下面是一个示例代码: ```python import numpy as np import pickle def load_stl10_data(path): data = {} data['train_X'] = [] data['train_y'] = [] data['test_X'] = [] data['test_y'] = [] data['unlabeled_X'] = [] with open(path+'/train_X.bin', 'rb') as f: data['train_X'] = np.fromfile(f, dtype=np.uint8).reshape((5000, 3, 96, 96)).transpose((0,3,2,1)).astype(np.float32) with open(path+'/train_y.bin', 'rb') as f: data['train_y'] = np.fromfile(f, dtype=np.uint8).astype(np.int32)-1 with open(path+'/test_X.bin', 'rb') as f: data['test_X'] = np.fromfile(f, dtype=np.uint8).reshape((8000, 3, 96, 96)).transpose((0,3,2,1)).astype(np.float32) with open(path+'/test_y.bin', 'rb') as f: data['test_y'] = np.fromfile(f, dtype=np.uint8).astype(np.int32)-1 with open(path+'/unlabeled_X.bin', 'rb') as f: data['unlabeled_X'] = np.fromfile(f, dtype=np.uint8).reshape((100000, 3, 96, 96)).transpose((0,3,2,1)).astype(np.float32) return data data = load_stl10_data('path/to/stl10') ``` 在此代码中,我们首先定义了一个空字典data,用于存储STL-10数据集的各个部分。然后,我们打开训练集、测试集和未标记的数据文件,并使用np.fromfile函数将其加载到NumPy数组中。由于数据是以二进制格式存储的,我们需要指定dtype=np.uint8。数据数组的形状为(num_samples,3,96,96),其中num_samples是样本数。我们需要将数组的维度重新排列为(num_samples,96,96,3)。最后,我们将数据类型转换为np.float32,并将标签减去1,以从0开始对标签进行索引。 4.使用数据集 现在,我们已经成功加载了STL-10数据集,我们可以使用它来训练模型或进行其他任务。例如,我们可以使用Matplotlib库来显示数据集中的图像: ```python import matplotlib.pyplot as plt # 显示训练集中的第一张图像 plt.imshow(data['train_X'][0]/255.) plt.show() ``` 这将显示训练集中的第一张图像。请注意,我们需要将像素值除以255,以将其缩放到0到1之间的范围内。 总结 在本教程中,我们学习了如何加载STL-10数据集,这是一个包含10个类别的图像数据集。我们使用Python中的NumPy和pickle模块来加载数据集,并使用Matplotlib库来显示图像。现在,我们可以使用STL-10数据集来训练模型或进行其他任务。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值