HDU_3038 How Many Answers Are Wrong 【带权并查集】

一、题面

HDU3038

二、分析

用并查集可以方便的判断两个位置是否有关系，这种关系可以通过是否有公共父节点判断，如果有公共父节点则可以直接判断是否正确，如果没有公共父节点，就可以把这个条件与之前的联系起来。同时需要设定sum,表示当前点到父节点的权值，这个权值方便后面的判断，这里有几种情况。

假设输入为$(x,y,w)$，那么对应父节点$fx = find(x), fy = find(y)$:

1.如果$fx==fy$：那么可以直接判断$w == sum[x] - sum[y]$.

2.如果$ x < y < fx < fy$：这里需要更新并查集的信息，$par[fx] = fy, sum[fx] = sum[y] - (sum[x] - w)$.

3.如果$ x < fx < y < fy$：此时，$par[fx] = fy, sum[fx] = sum[y] + (w - sum[x])$.

4.如果$ x < y < fy < fx$：此时，$par[fy] = fx, sum[fy] = sum[x] - sum[y] - w$.

分析上面四种情况，可以得出2和3可以合并，对于4,仔细分析一下， 相当于就是3的情况，只是因为$par[fy] = fx$，只是这样可以保证$sum$的值为正。注意在路径压缩时，对$sum$进行更新就可以了。

细心的朋友可能发现了，还有$x = y$的情况呢。对于这种情况，我们可以直接对输入的左值整体再往左移一下，就是输入的$u$变成$u-1$，这个对结果是没有影响的。但是如果不减，我们就要考虑$x=y$的情况了，前面我们已经初始化了$sum$初值为0，并且$x=y$的左右两边的情况我们是不能保证完全清楚的。所以这样把$x=y$变成了$(x-1,y]$这个区间的形式，就让问题又归到了我们上面讨论的情况，更简单了。

三、AC代码

 #include <cstdio>
 #include <iostream>
 #include <cstring>
 #include <fstream>
 #include <map>

 using namespace std;

 const int MAXN = 2e5+;
 int par[MAXN];
 int sum[MAXN];

 void Init()
 {
     for(int i = ; i < MAXN; i++)
         par[i] = i;
     memset(sum, , sizeof(sum));
 }

 int Find(int x)
 {
     if(par[x] == x)
         return x;
     int fx = par[x];
     par[x] = Find(fx);
     sum[x] = sum[fx] + sum[x];
     return par[x];
 }

 bool Union(int x, int y, int w)
 {
     int fx = Find(x);
     int fy = Find(y);
     // if(fx != fy)
     // {
     //     par[fx] = fy;
     //     sum[fx] = w + sum[y] - sum[x];
     //     return true;
     // }
     // else
     // {
     //     return w == sum[x] - sum[y];
     // }
     if(fx < fy)
     {
         par[fx] = fy;
         sum[fx] = w + sum[y] - sum[x];
         return true;
     }
     else if(fx > fy)
     {
         par[fy] = fx;
         sum[fy] = sum[x] - w - sum[y];
         return true;
     }
     else
     {
         return w == sum[x] - sum[y];
     }

 }

 int main()
 {
    // freopen("input.txt", "r", stdin);
     int N, M;
     while(scanf("%d %d", &N, &M)!=EOF)
     {
         Init();
         int a, b, w, ans = ;
         for(int i = ; i < M; i++)
         {
             scanf("%d %d %d", &a, &b, &w);
             if(!Union(a-, b, w))
                 ans++;
         }
         printf("%d\n", ans);
     }
     return ;
 }