并查集
并查集是寻找几个集合间的关系,可以查找他们是不是同属于一个集合中,也可以合并他们成为一个大集合。如果用人与人之间的关系来描述并查集的话,可以借用网上一个很好的博客。
我们可以假设这是各种门派,那么根据书中的关系,我们先确定每个门派的大哥,胡青牛的明教的大哥就是张无忌啦,宋青书武当派的老大是张三丰,丘处机的就是王重阳,周芷诺的是灭绝师太、段誉就是段正淳,玄难是虚竹,小龙女是杨过、而韦小宝和欧阳锋就是无门无派了。
我们规定一个老大数组,一开始每个人的数组都是自己,如何根据连线提供的关系就慢慢寻找自己门派的掌门人。比如胡青牛连着杨左使,那么胡青牛的老大数组就从自己变成了杨左使,然后杨左使的老大就是张无忌,以此类推,把所有的门派的关系都顺清楚。然后因为要一步一步找会有点麻烦,我们可以一提到胡青牛的老大就让他输出张无忌,而不是通过杨左使输出,我们就压缩这个找的过程,直接让张无忌的手下都是同级,而不是以前那种分层。
就像上图一样,这就是路径压缩。然后代码分成三部分,分别是初始化、找老大、合并门派。
代码如下
1 void init(){//先初始化每个人,让每个人的老大数组都是自己 2 for(int i = 1 ; i <= n ; i++){ 3 f[i] = i; 4 } 5 } 6 int getf(int i){//得到每个人的老大,如果他老大就是他自己就返回i,否则就继续寻找他的老大 7 if( f[i] != i ) 8 f[i] = getf(f[i]); 9 return f[i]; 10 } 11 void merge(int u,int v){//先找到u和v的老大、如果不相等就让右边的投入左边的门派,称为虚左原则 12 int t1 = getf(u); 13 int t2 = getf(v); 14 if(t1 != t2){ 15 f[t2] = t1; 16 } 17 }