题目链接:Problem - 3038
参考文献:带权并查集_syddf-CSDN博客_带权并查集
hdu 3038 How Many Answers Are Wrong (带权并查集入门)_F_cff的博客-CSDN博客
分析:
首先可以先参考此题目:并查集的应用 方法1.维护节点 到 根节点的距离 方法2:扩展与并查集: 奇偶游戏_啦啦啦32421的博客-CSDN博客
此题首先需要注意到的一点是使用带权并查集,这道题目最需要注意的一点就是这道题目是端到端的总和求和,而我们使用带权并查集的时候,是维护端到端的距离值。
举例:
[1,4] 的距离为30 , [3,4]的距离为10, 如果按照端到端的举例的话,含义就是1 ~ 4之间的距离是30,3~4的距离是10, 则[1,3]的距离是20,区间之间的关系是连续的。
但是此题则不是[1,4]的总和为30,[3,4]的总和为10,[1,2]的总和为20.区间之间的关系是离散的。
所以想要使用上面的带权并查集则需要将其变为一个开区间。
[1,4]的总和为30,则就是(0,4]的总和是30.[3,4]的总和为10,则就是(2,4]的总和为10.
那么旧可以知道(0,2]的距离是20.就将离散的区间变为了连续的区间。
上面这是最需要注意的一点。
然后便是带权并查集的思路了。
如果a,b不在同一个集合中,那么创建一个根节点,将a,b合并到同一个集合内,并且可以在这个集合内维护着a到根节点之间的距离dist[a],b到根节点之间的距离dist[b]。
通过dist[b]和dist[a]之间的关系,求出ab之间的距离。
所以可以知道,如果a,b已经在同一个集合中,那么直接求解出处理,然后判断是否相等,如果相等则结束。
如果不等则错误。
如果两个点不在同一个集合,那么一定可以构造一组数据,使得两个点之间这个距离满足条件,而我们按照上面的带权并查集,将其并入同一个集合即可。
同时还要注意,一定要固定根节点以小值作为根节点还是大值作为根节点。固定好根节点的关系,计算距离才更为容易。(比如如果根节点在这个集合中的中间,那么左边也有值,右边也有之,就无法判断左右两边值的关系。但是集合中的值在根节点的左边或者右边)。
代码实现:
# include <iostream>
using namespace std;
const int N = 200010;
int p[N];
int dist[N];
int n,m;
int find2(int x)
{
if(p[x] != x)
{
int temp = p[x];
p[x] = find2(p[x]);
dist[x] += dist[temp]; // 原dist[x]为x到temp的距离,dist[temp]现在改为了到新的根节点的距离
}
return p[x];
}
int main()
{
while(~scanf("%d %d",&n,&m))
{
for(int i = 0 ; i <= n ; i++)
{
p[i] = i;
dist[i] = 0; // 到根节点的距离
}
int ans = 0; // 错误的个数
for(int i = 1 ; i <= m ; i++)
{
int a,b,c;
scanf("%d %d %d",&a,&b,&c);
a--; // a-- 的原因是因为这道题目为端点,而我们的dist维护的是点到点之间的距离。
int la = find2(a);
int lb = find2(b);
if(la == lb) // 说明前面已经可以判断a,b的距离了
{
if(dist[b] - dist[a] != c) // a~b的距离不等于c
{
ans++;
}
}
else // la != lb,说明必定可以构造一组数据满足条件
{
//分两种情况
if(la < lb) // 如果a的根节点 小于 b的根节点
{
dist[lb] = c - dist[b] + dist[a];
p[lb] = la; // 并到小的节点上,或者并到大值节点上也是可以的。主要固定是小的作为根节点或者大的作为根节点即可
}
else if(la > lb)
{
dist[la] = dist[b] - dist[a] - c;
p[la] = lb;
}
}
}
printf("%d\n",ans);
}
return 0;
}