并查集。。。
第一次写带权并查集,感觉好难理解,看别人代码看了一上午才看懂,好弱菜的说。。。
所谓带权并查集,指的是在合并时不仅要将区间合并,而且要记录某些信息,记录的方法类似于数学的坐标系,以一个点为参考点(根节点),然后求出其他点到这个店的相对信息值,难点在于合并两个不同的集合时相对信息值的更新,分别在find和union里面进行更新,不理解的同学可以在纸上划一下,一边不懂多划几遍,表示我划了了两张纸才搞懂到底是如何更新如何维护的。。。
/*
带权并查集
*/
#include <iostream>
#include <cstring>
using namespace std;
int father[200005];
int rank[200005];
int ans;
int find(int x) //此时find不单有查找任务,还有更新距离任务
{
if (x==father[x])
return x;
int t=father[x];
father[x]=find(father[x]);
rank[x]+=rank[t]; //记录到根节点的距离,一定要有一个思想,根节点是一个区间的一个端点而不是一个区间,输入的区间被合并成了两个点
return father[x];
}
void Union(int a,int b,int s)
{
int x=find(a);
int y=find(b);
//cout<<x<<" "<<y<<endl;
//cout<<rank[a]<<" "<<s<<" "<<rank[b]<<endl;
if (x==y)
{
if (rank[a]+s!=rank[b])
ans++;
return ;
}
else
{
father[y]=x;
rank[y]=rank[a]+s-rank[b]; //y到x的距离等于a到x的距离+b到a的距离-b到y的距离
}
//cout<<a<<"\t\t"<<b<<endl;
//cout<<rank[a]<<"\t"<<x<<"\t"<<rank[b]<<endl;
}
void init()
{
memset(rank,0,sizeof(rank));
for (int i=0;i<=200003;i++)
father[i]=i;
}
int main()
{
int n,m;
while (cin>>n>>m)
{
int a,b,s;
ans=0;
init();
for (int i=1;i<=m;i++)
{
cin>>a>>b>>s;
Union(a,b+1,s);
}
cout<<ans<<endl;
}
}