hdu3038（并查集）

题目链接

题意：有n次询问，给出a到b区间的总和，问这n次给出的总和中有几次是和前面已近给出的是矛盾的

分析：sum数组维护着到根节点的距离（即区间和），每次合并x,y,s(a,b分别为x,y的根节点)时（假设由a指向b），可根据sum[x](x到达a的距离)，sum[y]（y到达b的距离）,s(x到达y的距离)，然后推出x的根节点a到达y的根据点b的距离，剩下x到达y的根节点a的距离，再在find函数里再更新。

#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <queue>
#include <cstdlib>
#include <stack>
#include <vector>
#include <set>
#include <map>
#define LL long long
#define mod 100000000
#define inf 0x3f3f3f3f
#define eps 1e-9
#define N 200010
#define FILL(a,b) (memset(a,b,sizeof(a)))
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define PII pair<int,int>
using namespace std;
int fa[N],vis[N],sum[N];
int find(int x)
{
    if(x==fa[x])return x;
    int pa=fa[x];
    fa[x]=find(fa[x]);
    //当父节点pa指向新的根节点时，sum[x]是x到达父节点pa的距离，pa到达的新根节点之前已更新好
    //因此sum[x]+sum[pa]就是x到达新根节点的距离了
    sum[x]=sum[x]+sum[pa];
    return fa[x];
}
void merge(int x,int y,int s)
{
    int a=find(x);
    int b=find(y);
//sum[y]表示y到b的距离，sum[x]表示x到a的距离，s表示x到y的距离，模拟一下得出a到b的距离或b到a的距离
    if(a>b)
    {
        fa[b]=a;
        sum[b]=sum[x]-s-sum[y];
    }
    else
    {
        fa[a]=b;
        sum[a]=sum[y]+s-sum[x];
    }
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)>0)
    {
        for(int i=0;i<=n;i++)
        {
            fa[i]=i;sum[i]=0;
        }
        int ans=0;
        for(int i=1;i<=m;i++)
        {
            int x,y,s;
            scanf("%d%d%d",&x,&y,&s);
            x--;
            if(find(x)==find(y))
            {
                if(sum[x]!=sum[y]+s)
                ans++;
            }
            else merge(x,y,s);
        }
        printf("%d\n",ans);
    }
}