POJ 3038 How Many Answers Are Wrong（加权并查集）

题目链接：
POJ 3038 How Many Answers Are Wrong

题意：
n表示n个数字，编号1–n，然后有m个区间[l,r]和该区间和s，
问在这m个区间中有多少个区间和是不正确的？
如果不正确就忽略该区间和，否则将该区间和作为已知条件使用。

分析：
需要一个数组val,val[i]表示i到根节点的距离，然后就是在查找根节点的过程更新路径上结点的val,和在mix函数里判断该区间和是否有效。
l到r之间的和为s可以理解为l-1到r的距离为s。

#include <iostream>#include <cstdio>#include <cmath>#include <algorithm>#include <cstring>#include <string>#include <stack>#include <queue>using namespace std;const int maxn = 200010;int val[maxn], pre[maxn];//val[i]:i到根节点的距离；pre[i]：i的父节点int n, m, u, v, w;int find(int x)//递归寻找x的根节点{    if (pre[x] == x) return x;    int tmp = find(pre[x]);//tmp是x的根节点    val[x] = val[x] + val[pre[x]];    //在递归时x还未连接到根节点上，只连接到父节点上，所以这时的val[x]实际上是到父节点的距离    //而val[pre[x]]是父节点到根节点的距离，所以真正的val[x]=val[x]+val[pre[x]]，左边的val[x]是到根节点距离，右边的是到父节点距离    //可以从递归倒数第二层往前想    return pre[x] = tmp;//路径压缩}int mix(int x, int y, int z){    int fx = find(x);    int fy = find(y);    if (fx != fy)    {        pre[fx] = fy;//将x的根节点fx的父节点设为fy        val[fx] = val[y] - val[x] + z;        //x到y的距离是z，x到fx的距离是val[x]，y到fy的距离是val[y],        //那么fx到fy的距离val[fx]=x到fy的距离（z+val[y]）- x到fx的距离（val[x]）        return 0;    }    else    {        if (abs(val[y] - val[x]) == z) return 0;//必须是绝对值才行        //因为尽管y>x，但是到根节点的距离不确定大小（到根结点路径上个点的值大小不确定）        return 1;    }}int main(){#ifdef LOCAL    freopen("in.txt", "r", stdin);    //freopen("out.txt", "w", stdout);#endif    while (~scanf("%d%d", &n, &m))    {        for (int i = 0; i <= n; i++)//因为接下来由u--的存在，所以要从0开始            pre[i] = i;        memset(val, 0, sizeof(val));        int ans = 0;        for (int i = 1; i <= m; i++)        {            scanf("%d%d%d", &u, &v, &w);            u--;//这样做符合val的含义            ans += mix(u, v, w);        }        printf("%d\n", ans);    }    return 0;}