hdu 3038 How Many Answers Are Wrong

点击打开链接hdu 3038

思路: 带权并查集
分析:
1 给定一序列的区间的和，求错误的个数
2 典型的带权并查集，区间[l,r]的和等于sum[r]-sum[l-1]。对于一般涉及到区间和还有个数的问题的时候，都要把左下标减一来处理
3 用rank[x]表示的是x到跟节点的和即[x,find(x)]的和，那么对于给定的x,y,val，fx为x的跟节点,fy为y的跟节点
如果fx != fy那么这个时候就要考虑fx和fy的关系
fx > fy ，则father[fy] = fx; rank[fy] = rank[x]-val-rank[y];
fx < fy ，father[fx] = fy; rank[fx] = rank[y]+val-rank[x];
否则判断rank[x]-rank[y]是否为val

代码:

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

const int MAXN = 200010;

int father[MAXN];
int rank[MAXN];

void init(int n){
    memset(rank , 0 , sizeof(rank));
    for(int i = 0 ; i <= n ; i++)
        father[i] = i;
}

int find(int x){
    if(father[x] != x){
       int fa = father[x];
       father[x] = find(fa);
       rank[x] += rank[fa];
    }
    return father[x];
}

bool solve(int x , int y , int val){
    int fx = find(x);
    int fy = find(y);
    if(fx != fy){
        if(fx > fy){
            father[fy] = fx;
            rank[fy] = rank[x]-val-rank[y];
        }
        else{
            father[fx] = fy;
            rank[fx] = rank[y]+val-rank[x];
        }
    }
    else{
        return rank[x]-rank[y] == val;
    }
    return true;
}

int main(){
    int n , m , ans;
    int x , y , val;
    while(scanf("%d%d" , &n , &m) != EOF){
         init(n);
         ans = 0;
         while(m--){
              scanf("%d%d%d" , &x , &y , &val);
              if(!solve(x-1 , y , val))
                  ans++;
         }
         printf("%d\n" , ans);
    }
    return 0;
}