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);

    }

}

View Code