题意: 一圈座位有n个,给出m组序号之间的关系,比如,1 2 150 代表2号坐在1号位置序号+150,看m组数据有多少组冲突的。
思路: 带权并查集模板。
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
#include<queue>
#include<vector>
#include<stdlib.h>
using namespace std;
#define N 50010
int f[N],n,m,val[N],cnt;
int getf(int v)
{
if(f[v]==v)
return v;
int t=f[v];
f[v]=getf(f[v]);/*压缩路径*/
val[v]+=val[t];/*记录当前节点到祖先节点的权值和(在压缩路径的过程中,遍历到他的祖先,加上权值)*/
return f[v];
}
void marge(int u,int v,int s)
{
int t1=getf(u),t2=getf(v);
if(t1!=t2)
{
f[t2]=t1;/*搞清楚节点之间的父子关系!!!*/
val[t2]=val[u]-val[v]+s;/*向量*/
}
else if(val[v]!=val[u]+s)/*祖先相同,之间距离关系也应该正确,否则矛盾*/
cnt++;
}
int main()
{
while(~scanf("%d%d",&n,&m))
{
cnt=0;
for(int i=1; i<=n; i++)
f[i]=i,val[i]=0;
int v,u,s;
for(int i=1; i<=m; i++)
{
scanf("%d%d%d",&u,&v,&s);
marge(u,v,s);
}
printf("%d\n",cnt);
}
return 0;
}