HDU 3038

带权并查集

并查集的入门知识可以百度：并查集有趣篇

没接触过这样题目，本小菜不会。所以这里写详细一点，算是做个笔记

题意：有N个数和M条信息，没条信息包含3个数 a b c ，表示第 a 个数到第 b 个数的和为 c 。

如果第 i 条信息与前 i-1 条信息没有冲突，则认为它是真的，否则是假的。问假信息有多少条。

怎么用并查集解的呢？首先做一点小处理，将 a-- 。为什么要这样呢，题目给的是一个闭区间[a,b]的和

把它换成一个半开半闭区间 (a--,b] 。半开半闭区间有个优良的性质： (a,b] + (b,c] = (a,c]

再来看这 M 条信息，对于一条信息 a,b,c 怎么解读它呢？我们可以把它解读为：a与b有关系，它们的关系有一个权值c

将关系用并查集的方式存储下来，那么对于处于同一个集合里面的元素两两之间的关系权值要么已知，要么可以推算出来。

原因很简单  (a,b] + (b,c] = (a,c] 。那么当新来一条信息 ai,bi,ci 时，判断 ai,bi 是不是在一个集合里。如果不在，将 ai,bi所处

的集合合并；如果在，判断一下 ci 是否等于 ai,bi 的关系权值。

现在，大体上的思路已经明了。剩下的问题是，怎么去维护关系权值呢？两两去算去存显然很麻烦，时间空间复杂度也不允许。

这些权值是建立在关系上的，还得让并查集来帮忙。建立一个weight[]数组，weight[a] 表示 a 与它当前的老大per[a]的关系权值。

如果说 a,b 有关系，那么 a,b 间的关系权值就等于 weight[b] - weight[a] （这里假设b>a）

到此为止，思路上的问题已经全部解决了。编码上，核心代码就是并查集的插入，压缩路径，wight数组的维护在插入和压缩路径时处理，详见代码。

#include<stdio.h>
#include<iostream>
#include<memory.h>
using namespace std;

#define Maxn 200010

int n,m,per[Maxn],weight[Maxn],ans;

void join(int fx,int fy,int x,int y,int w);
int find(int x);
int main()
{
    int a,b,c,fa,fb;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(weight,0,sizeof(weight));
        ans=0;
        for(int i=0;i<=n;i++)
            per[i]=i;

        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&a,&b,&c);
            a--;
            fa=find(a),fb=find(b);
            if(fa==fb)
            {
                if(weight[b]-weight[a]!=c)
                    ans++;
            }
            else
            {
                join(fa,fb,a,b,c);
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}

//  fx <---- fy
//   ^        ^
//   |        |
//   x        y
//新增了fx<---fy，所以要维护weight[fy]，满足等式
//weight[y]+weight[fy]-weight[x]=w
void join(int fx,int fy,int x,int y,int w)
{
    per[fy]=fx;
    weight[fy]=weight[x]-weight[y]+w;
}
int find(int x)     //查询的同时压缩路径，采用递归的方式
{                   //从最接近老大的节点开始维护weight
    if(per[x]==x)
        return x;

    int r=find(per[x]);
    weight[x]+=weight[per[x]];
    per[x]=r;
    return r;
}