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

题目链接http://www.lydsy.com/JudgeOnline/problem.php?id=4690
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 ，表示不能回答问题。
Sample Input

2 2

! 1 2 1

? 1 2

2 2

! 1 2 1

? 2 1

4 7

! 1 2 100

? 2 3

! 2 3 100

? 2 3

? 1 3

! 4 3 150

? 4 1

0 0
Sample Output

1

-1

UNKNOWN

100

200

-50

题目大意：n个数，m个关系，有两种操作：给出x,y,k表示x比y小k；询问x比y大多少

裸的加权并查集
v[i]表示i节点比i的祖先节点小多少，则合并时v[fx]=v[y]+k-v[x]，v[x]=v[x]+v[fa[x]]即可…


代码如下:

#include<bits/stdc++.h>
#define N 100050
using namespace std;
int n,m,x,y,k,f[N],v[N];
char s[5];
int find(int x){
    if(x==f[x]) return x;
    int tmp;
    f[x]=find(tmp=f[x]);
    v[x]=v[x]+v[tmp];
    return f[x];
}
int main(){
    while(~scanf("%d%d",&n,&m)&&n&&m){
        for(int i=1;i<=n;i++) f[i]=i,v[i]=0;
        for(int i=1;i<=m;i++){
            scanf("%s%d%d",s,&x,&y);
            if(s[0]=='!'){
                scanf("%d",&k);
                int fx=find(x),fy=find(y);
                f[fx]=fy;
                v[fx]=v[y]+k-v[x];
            }
            else{
                int fx=find(x),fy=find(y);
                if(fx!=fy) printf("UNKNOWN\n");
                else printf("%d\n",v[x]-v[y]);
            }
        }
    }
return 0;
}