HDU 3038 How Many Answers Are Wrong(并查集)

转载请注明出处，谢谢/article/2566293.html
by---cxlove

题目：给出一些询问，[l,r]的和为s，问有多少个是错的

http://acm.hdu.edu.cn/showproblem.php?pid=3038

我们考虑sum[i]表示i与根节点之间节点的和

也就是sum[i]=sigma(aj) (pre[i]=>j>=i)

另外需要注意的是sum[r]-sum[l-1]表示的是闭区间[l,r]的和

最近好水啊~~~

#include<iostream>
#include<cstdio>
#include<map>
#include<cstring>
#include<cmath>
#include<vector>
#include<algorithm>
#include<set>
#include<string>
#include<queue>
#define inf 1<<30
#define M 60005
#define N 10005
#define maxn 300005
#define eps 1e-10
#define zero(a) fabs(a)<eps
#define Min(a,b) ((a)<(b)?(a):(b))
#define Max(a,b) ((a)>(b)?(a):(b))
#define pb(a) push_back(a)
#define mem(a,b) memset(a,b,sizeof(a))
#define LL long long
#define lson step<<1
#define rson step<<1|1
#define MOD 1000000009
#define sqr(a) ((a)*(a))
using namespace std;
int n,m;
int pre[200005];
int sum[200005];
int find(int x)
{
    if(x!=pre[x])
    {
        int f=pre[x];
        pre[x]=find(pre[x]);
        sum[x]+=sum[f];
    }
    return pre[x];
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=0;i<=n;i++) pre[i]=i,sum[i]=0;
        int ans=0;
        while(m--)
        {
            int l,r,s;
            scanf("%d%d%d",&l,&r,&s);
            l--;
            int ra=find(l),rb=find(r);
            if(ra==rb)
            {
                if(sum[l]-sum[r]!=s) ans++;
            }
            else
            {
                pre[ra]=rb;
                sum[ra]=sum[r]-sum[l]+s;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}