[数据结构] 并查集

相关概念

• 初始时，每个数据的下标都为-1，表示10棵树；然后采用双亲表示法，一个位置值为负，那他就是树的根，负数的绝对值表示这棵树的数据个数；一个位置值为正，该正数就是他双亲的下标。
• 如图所示，假设有10个数据要进行并查集，具体做法如下：
  其中，数组的下标对应集合中元素的编号
  

并查集的模拟实现

1）实现基本框架

• 其底层是vector数组，初始时各元素的值为-1

template <class T>
struct union_find_set_2{
public:
	union_find_set_2(size_t n){
		_array.resize(n, -1);  //n个元素都初始化为-1，表示n棵树
	}
	private:
	vector<T> _array;
};

2）实现基础操作

findRoot查找元素属于哪个集合

	int findRoot(int index){
		while (_array[index] >= 0){  
			index = _array[index];
		}
		return index;
	}

Union合并两个集合

void Union(int index1, int index2){
		int root1 = findRoot(index1);
		int root2 = findRoot(index2);
		if (root1 == root2)
			return;
		else{
			_array[root1] += _array[root2];
			_array[root2] = root1;
		}
	}

IsOneSet判断两个元素是否属于同一集合

bool isOneSet(int index1, int index2){
		int root1 = findRoot(index1);
		int root2 = findRoot(index2);
		return root1 == root2;
	}

SetSize集合个数

size_t setSize(){
		size_t count = 0;
		for (size_t i = 0; i < _array.size(); i++){
			if (_array[i] <= -1){
				count++;
			}
		}
		return count;
	}