[BZOJ]4690: Never Wait for Weights 带权并查集

Description

在实验室中，Nathan Wada作为助手的职责是测定两个样品的重量差异。当样品的差异很小时，使用天平能比使用

弹簧秤得到更精确的结果，所以他只使用天平来测得一些样品的重量差。他偶尔会被询问一些样品的重量差，而他

能否回答这些问题取决于在回答相应问题时他已经得到的测量结果。由于他所在处理的测量数据是巨大的，所以他

希望你能写个程序帮他处理数据和回答问题。

Input

输入包含多组测试数据。每组数据第一行包含两个整数 N 和 M ，其中 N 表示样品的数量，

样品从 1 到 N 标号，满足 2≤N≤100000 。

接下来 M 行，每行包括一个测量结果或者询问，按时间顺序给出，满足 1≤M≤100000 。

一个测量结果被格式化为 ! a b w ，表示第 a 个样品比第 b 个样品轻 w 个单位重量

满足 a≠b,0≤w≤1000000 ，且任意的测试结果互不矛盾。

一个询问被格式化为 ? a b ，表示询问第 a 个样品比第 b 个样品轻多少个单位重量，满足 a≠b 。

输入以两个零作为结束。

Output

对于每个询问输出一行，如果能回答问题，则输出问题的答案，你可以认为答案的绝对值不超过 1000000 

否则输出 UNKNOWN ，表示不能回答问题。

第一道带权并查集的题目，还挺好懂的，注意下细节，多复习……

代码：

#include<cstdio>#include<cstring>const int maxn=100005;int f[maxn],d[maxn];int findfa(int x){if(f[x]==x) return x;//d[x]+=d[f[x]];int oldfa=f[x];f[x]=findfa(f[x]);d[x]+=d[oldfa];return f[x];}void work(int x,int y,int z){int fx=findfa(x),fy=findfa(y);//if(fx==fy) return;f[fx]=fy;d[fx]=d[y]+z-d[x];}int main(){int n,m;while(1){scanf("%d%d",&n,&m);if(n==0 && m==0) break;for(int i=1;i<=n;i++) f[i]=i,d[i]=0;for(int i=1;i<=m;i++){char s[3];int x,y,z;scanf("%s%d%d",s,&x,&y);if(s[0]=='!'){scanf("%d",&z);work(x,y,z);}else {int fx=findfa(x),fy=findfa(y);if(fx!=fy) printf("UNKNOWN\n");else printf("%d\n",d[x]-d[y]);}}}}