【 HDU 3038 How Many Answers Are Wrong】 并查集好题

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3038

题目大意：

给你n给整数，m次操作。

每次操作是给出三个整数a，b，c； 子序列a~b的和为c。

找出m次操作中有多少次不合法，不合法的操作舍去。

 

解题思路：

先开一个dis[n]数组，dis[x]记录的是与父节点的距离。

d[x]记录的是[0，x]的和    

dis[x]=d[x]-d[pre[x]]；

对于每个操作：

1.如果操作合法，则进行更新。 

  注意：节点小的作为父节点放前面。

2.  如果不合法（前提是父节点相同），即dis[b]-dis[a]不等于c，次数加1.

 

#include <iostream>
#include <cstring>
#include <cstdio>
#include <map>
#include <algorithm>
using namespace std;

const int maxn=200005;
int pre[maxn];
int dis[maxn];

int find(int x)
{
    int r=x;
    if(pre[x]!=x)
    {
        r=pre[x];
        pre[x]=find(pre[x]);
        dis[x]+=dis[r];
    }
    return pre[x];
}

int main()
{
    int n, m, fa, fb, u, v, c, disa, disb;
    while(cin >> n >> m)
    {
        for(int i=0; i<=n; i++)
            pre[i]=i;
        memset(dis,0,sizeof(dis));
        int ans=0;
        for(int i=0; i<m; i++)
        {
            scanf("%d%d%d",&u,&v,&c);
            u--;
            fa=find(u);
            fb=find(v);
            if(fa!=fb)
            {
                if(fa<fb)
                {
                    dis[fb]=dis[u]-dis[v]+c;
                    pre[fb]=fa;
                }
                else
                {
                    dis[fa]=dis[v]-dis[u]-c;
                    pre[fa]=fb;
                }
            }
            else
            {
                disa=dis[u];
                disb=dis[v];
                if(disb-disa!=c)
                    ans++;
            }
        }
        cout << ans <<endl;
    }
    return 0;
}