带权并查集
并查集的入门知识可以百度:并查集有趣篇
没接触过这样题目,本小菜不会。所以这里写详细一点,算是做个笔记
题意:有N个数和M条信息,没条信息包含3个数 a b c ,表示第 a 个数到第 b 个数的和为 c 。
如果第 i 条信息与前 i-1 条信息没有冲突,则认为它是真的,否则是假的。问假信息有多少条。
怎么用并查集解的呢?首先做一点小处理,将 a-- 。为什么要这样呢,题目给的是一个闭区间[a,b]的和
把它换成一个半开半闭区间 (a--,b] 。半开半闭区间有个优良的性质: (a,b] + (b,c] = (a,c]
再来看这 M 条信息,对于一条信息 a,b,c 怎么解读它呢?我们可以把它解读为:a与b有关系,它们的关系有一个权值c
将关系用并查集的方式存储下来,那么对于处于同一个集合里面的元素两两之间的关系权值要么已知,要么可以推算出来。
原因很简单 (a,b] + (b,c] = (a,c] 。那么当新来一条信息 ai,bi,ci 时,判断 ai,bi 是不是在一个集合里。如果不在,将 ai,bi所处
的集合合并;如果在,判断一下 ci 是否等于 ai,bi 的关系权值。
现在,大体上的思路已经明了。剩下的问题是,怎么去维护关系权值呢?两两去算去存显然很麻烦,时间空间复杂度也不允许。
这些权值是建立在关系上的,还得让并查集来帮忙。建立一个weight[]数组,weight[a] 表示 a 与它当前的老大per[a]的关系权值。
如果说 a,b 有关系,那么 a,b 间的关系权值就等于 weight[b] - weight[a] (这里假设b>a)
到此为止,思路上的问题已经全部解决了。编码上,核心代码就是并查集的插入,压缩路径,wight数组的维护在插入和压缩路径时处理,详见代码。
#include<stdio.h>
#include<iostream>
#include<memory.h>
using namespace std;
#define Maxn 200010
int n,m,per[Maxn],weight[Maxn],ans;
void join(int fx,int fy,int x,int y,int w);
int find(int x);
int main()
{
int a,b,c,fa,fb;
while(scanf("%d%d",&n,&m)!=EOF)
{
memset(weight,0,sizeof(weight));
ans=0;
for(int i=0;i<=n;i++)
per[i]=i;
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&a,&b,&c);
a--;
fa=find(a),fb=find(b);
if(fa==fb)
{
if(weight[b]-weight[a]!=c)
ans++;
}
else
{
join(fa,fb,a,b,c);
}
}
printf("%d\n",ans);
}
return 0;
}
// fx <---- fy
// ^ ^
// | |
// x y
//新增了fx<---fy,所以要维护weight[fy],满足等式
//weight[y]+weight[fy]-weight[x]=w
void join(int fx,int fy,int x,int y,int w)
{
per[fy]=fx;
weight[fy]=weight[x]-weight[y]+w;
}
int find(int x) //查询的同时压缩路径,采用递归的方式
{ //从最接近老大的节点开始维护weight
if(per[x]==x)
return x;
int r=find(per[x]);
weight[x]+=weight[per[x]];
per[x]=r;
return r;
}