带权并查集
之前做过几乎一样的题目(这里),然而忘光了
记 wi w i 表示根节点到 i i 这段时间的盈利,当和 y y 不属于同一个根节点时,更新并合并。当属于同一个根节点时判断 wy−wx w y − w x 是否 =z = z 。
代码:
#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 105
using namespace std;
inline char readc(){
static char buf[100000],*l=buf,*r=buf;
if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
return l==r?EOF:*l++;
}
inline int _read(){
int x=0,f=1; char ch=readc();
while (!isdigit(ch)) { if (ch=='-') f=-1; ch=readc(); }
while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
return x*f;
}
int fa[N],w[N];
int findfa(int x){
if (fa[x]==x) return x; int f=findfa(fa[x]);
w[x]+=w[fa[x]]; return fa[x]=f;
}
int main(){
for (int t=_read();t;t--){
int n=_read(),m=_read(); bool f=true;
for (int i=0;i<=n;i++) fa[i]=i,w[i]=0;
while (m--){
int x=_read()-1,y=_read(),z=_read(),fx,fy;
if ((fx=findfa(x))!=(fy=findfa(y)))
w[fy]=z+w[x]-w[y],fa[fy]=fx;
else if (w[y]-w[x]!=z) f=false;
}
puts(f==true?"true":"false");
}
return 0;
}