数据结构——并查集

算法： 并查集（union & find）

用途：等价类的处理，最后期限任务调度。

 

最近开始复习数据结构。

看的是《数据结构算法与应用-c++语言描述》。

第三章：数据描述

看到后面的等价类，想到了以前用的并查集。

发现以前对于UFSET的理解是比较浅薄的。

于是乎，在网上找了些资料，然后又在PKU上找了些题目来练习下。

 

 

网上提到了“食物链”这个典型的题目。

记得以前看了好多遍就是不怎么明白，今天看了一份报告是用向量来解释的，非常好，易于理解。

 

const int MAX_SIZE = 1002;

class CUFSet

{

public:

    CUFSet(int size)

    {

        nSize = size;

        for (int i = 0; i < nSize; i ++)

        {

            parent[i] = -1; // parent[i] 为负表示是根节点，同时它的绝对值表示已该节点为根的集合中的元素个数（上次比赛时竟然没想出来。。。。）

            relation[i] = 0; // 用于处理向量关系

        }

    }

    void init(int n)

    {

        for (int i = 0; i <= n; i ++)

        {

            parent[i] = -1;

            relation[i] = 0;

        }

    }

    int Find(int x);

    void Union(int rootx, int rooty);

    //

    int getRelation(int x)

    {

        return relation[x];

    }

private:

    int parent[MAX_SIZE];

    int relation[MAX_SIZE]; //the relation with the root node: 0 same, 1 different

    int nSize;

};

int CUFSet::Find(int x)

{

    if (parent[x] < 0)

    {

        return x;

    }

    int xParent = parent[x];

    parent[x] = Find(parent[x]); //路径压缩

//relation[x] = (relation[x] + relation[xParent])%2; //路径压缩时跟新关系之

    return parent[x];

}

void CUFSet::Union(int rootx, int rooty)

{

    if (rootx == rooty)

    {

        return;

    }

    int tmp = parent[rootx] + parent[rooty]; //加权

    if (parent[rootx] < parent[rooty])

    {

        parent[rooty] = rootx;

        parent[rootx] = tmp;

    }

    else

    {

        parent[rootx] = rooty;

        parent[rooty] = tmp;

    }

}