标签:
想法好像比较奇怪……打死我也没想到这是个并查集啊。
并查集维护两个数组,f[i]表示i所在的并查集,v[i]表示f[i]->i月的收入,每次读入s,t,令p=f[s],q=f[t],若p==q,说明s,t属于同一个并查集,找f[s],f[t]的过程中维护处v[s],v[t],flag=v[t+1]-v[s]==w?1:0。若p!=q,则将f[q]赋值为p,(令每个并查集的左端点作为父亲结点,维护v[q])。
另:由于维护的是前缀和,所以读入是可将t+1,判断v[t+1]-v[s]的值。
#include<iostream> #include<cstdio> #include<cstdlib> #include<cstring> #include<algorithm> #include<cmath> using namespace std; int f[110],v[110]; int n,m,s,t,w,ww; bool flag; int find(int x) { if (f[x]==x) return x; int pp=f[x]; f[x]=find(f[x]); v[x]+=v[pp]; return f[x]; } int main() { scanf("%d",&w); while (w--) { memset(v,0,sizeof(v)); flag=0; scanf("%d%d",&n,&m); for (int i=1;i<=n+1;i++) f[i]=i; for (int i=1;i<=m;i++) { scanf("%d%d%d",&s,&t,&ww); t++; int p=find(s),q=find(t); if (p==q) { if (v[t]-v[s]!=ww) { flag=1; break; } } else { f[q]=p; v[q]=v[s]+ww-v[t]; } } if (flag) printf("false\n"); else printf("true\n"); } return 0; }
标签:
原文地址:http://www.cnblogs.com/ws-fqk/p/4658769.html