B - Zjnu Stadium（HDU - 3047）带权并查集

这道题是指一个圆形操场有300列，有以下n个人要坐在操场上，这n个人的坐法有m种限制。

以下m行每行有3个数 x,y,z,   是指从x到y中间要隔z列座位。 没有限制坐哪，但同一列只能坐一个人。

输出最后有几种限制是错的。（如果当前的限制条件与前面的限制条件相违背，就是错的）

这道题用并查集写，

当有两个人的座位发出来的时候，将它们和并到一个集合里，如果发现已经合并到一个集合里，说明他们都单独被搜过，此时比较二者之间的距离，如果不等于给出的限制条件，就是错的。

带权并查集是指在搜的过程中，并不只单独将他们合并到一个集合里，而是用一个数组记录他们之间的关系。这道题就是这样。

在将他们合并到一起的时候更新，同样用了路径压缩；

int add(int x)
{
    if(x==pre[x])
        return x;
    int root=pre[x];
    pre[x]=add(pre[x]);
    dis[x]+=dis[root];
    return pre[x];
}

如上图所示，pre数组记录着他们的祖先，dis数组记录了他们离祖先的距离。

如果当时二者处于同一个祖先中，dis[root]的值应为0，因为root就是他们的祖先，祖先距离祖先的位置就是0，

如果二者不处于同一个祖先中，说明至少有一个数没有被搜过，没被搜过的祖先是自己，（他们的初值都是0）；

root为x的上一步的祖先，里面dis[root]记录着root到祖先的长度，root为另一个集合的祖先，当两个集合合并后root就是x的父节点。

所以x到同一个祖先的长度为dis[x]+dis[root],，

void join(int x,int y,int s)
{
    int fx=add(x),fy=add(y);
    if(fx!=fy)
    {
        pre[fy]=fx;
        dis[fy]=dis[x]+s-dis[y];
    }
    else
    {
        if(dis[y]-dis[x]!=s)
        {
            cnt++;
        }
    }
}

 

pre[fy]=fx;这一步将fy的祖先命名为fx，

dis[fy]=dis[x]+s-dis[y];这条语句是将此时的dis[x]和dis[y]的差值更新为s,因为执行过pre[fy]=fx这条语句后，dis[y]-dis[x]就应该为s了，但此时还不一定相等，所以将dis[fy]的值更新为dis[x]+s-dis[y]，就是fy距离fx的值，也是dis[x]和dis[y]之间的差值，将fy和fx合并后，再进行更新。到查询的时候加上他们祖先的dis值，（因为路径压缩，此时找到的root为未更新的父节点，即更新前的祖先，也就是fy，返回值才为更新后的祖先）就是距离同一个祖先的距离。

代码如下

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;

int pre[50010],dis[50010],cnt;
int add(int x)
{
    if(x==pre[x])
        return x;
    int root=pre[x];
    pre[x]=add(pre[x]);
    dis[x]+=dis[root];
    return pre[x];
}
void join(int x,int y,int s)
{
    int fx=add(x),fy=add(y);
    if(fx!=fy)
    {
        pre[fy]=fx;
        dis[fy]=dis[x]+s-dis[y];
    }
    else
    {
        if(dis[y]-dis[x]!=s)
        {
            cnt++;
        }
    }
}
int main()
{
    int n,m,a,b,c,i;
    while(scanf("%d %d",&n,&m)!=EOF)
    {
        cnt=0;
        for(i=1;i<=n;i++)
        {
            pre[i]=i;
            dis[i]=0;
        }
        while(m--)
        {
            scanf("%d %d %d",&a,&b,&c);
            join(a,b,c);
        }
        printf("%d\n",cnt);
    }
    return 0;
}