hdu 3038：How Many Answers Are Wrong

TT和FF玩猜数游戏。一个未知数列，FF给出区间[a , b]，TT给出该范围内的数字和。但有时TT会给出错误答案。给定数列长度n与区间数m，之后每组给出区间范围a , b与TT给出的和s，求错误答案的数量。测试数据不止一组。

 

用前缀和的思想来考虑数列，于是可以用带权并查集来做。注意做的时候虽然每组给出的范围为a~b，求的时候要用(a-1)~b，这样才是前缀和。

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

const int MAXN = 200010 ;
int fa[MAXN] , sum[MAXN] ;
int WA ;

int get_root(int x) {
    if (x == fa[x]) return x ;

    int temp = fa[x] ;
    fa[x] = get_root(fa[x]) ;
    sum[x] += sum[temp] ;
    return fa[x] ;
}

void Union(int a , int b , int s) {
    int root_a , root_b ;
    root_a = get_root(a) ;
    root_b = get_root(b) ;
    if (root_a == root_b) {
        if (sum[a] + s != sum[b])
            WA ++ ;
    }
    else {
        fa[root_b] = root_a ;
        sum[root_b] = sum[a] + s - sum[b] ;
    }
}

int main() {
    //freopen("in.txt" , "r" , stdin) ;
    int n , m ;
    while (cin >> n >> m) {

        WA = 0 ;
        for (int i = 0 ; i <= n ; i ++) {
            fa[i] = i ; sum[i] = 0 ;
        }

        while (m --) {
            int a , b , s ;
            cin >> a >> b >> s ;
            a -- ;
            Union(a , b , s) ;
        }

        cout << WA << endl ;
    }
    return 0 ;
}