并查集

#include <stdlib.h>/** 并查集. */typedef struct ufs_t {    int *p; /** 树的双亲表示法*/    int size; /** 大小. */} ufs_t;/*** @brief 创建并查集.* @param[in] n 数组的容量* @return 并查集*/ufs_t* ufs_create(int n) {    ufs_t *ufs = (ufs_t*)malloc(sizeof(ufs_t));    int i;    ufs->p = (int*)malloc(n * sizeof(int));    for(i = 0; i < n; i++)        ufs->p[i] = -1;    return ufs;}/*** @brief 销毁并查集.* @param[in] ufs 并查集* @return 无*/void ufs_destroy(ufs_t *ufs) {    free(ufs->p);    free(ufs);}/*** @brief Find 操作，带路径压缩，递归版.* @param[in] s 并查集* @param[in] x 要查找的元素* @return 包含元素x 的树的根*/int ufs_find(ufs_t *ufs, int x) {    if (ufs->p[x] < 0) return x; // 终止条件        return ufs->p[x] = ufs_find(ufs, ufs->p[x]); /* 回溯时的压缩路径*/}/** Find 操作，带路径压缩，迭代版.所谓的路径压缩，就是在每次查找时，令查找路径上的每个节点都直接指向根节点 */static int ufs_find_iterative(ufs_t *ufs, int x) {    int oldx = x; /* 记录原始x */    while (ufs->p[x] >= 0) {        x = ufs->p[x];    }    while (oldx != x) {        int temp = ufs->p[oldx];        ufs->p[oldx] = x;        oldx = temp;    }    return x;}/*** @brief Union 操作，将y 并入到x 所在的集合.* @param[in] s 并查集* @param[in] x 一个元素* @param[in] y 另一个元素* @return 如果二者已经在同一集合，并失败，返回-1，否则返回0*/int ufs_union(ufs_t *ufs, int x, int y) {    const int rx = ufs_find(ufs, x);    const int ry = ufs_find(ufs, y);    if(rx == ry) return -1;    ufs->p[rx] += ufs->p[ry];    ufs->p[ry] = rx;    return 0;}/*** @brief 获取元素所在的集合的大小* @param[in] ufs 并查集* @param[in] x 元素* @return 元素所在的集合的大小*/int ufs_set_size(ufs_t *ufs, int x) {    const int rx = ufs_find(ufs, x);    return -ufs->p[rx];}