图论训练 重量差异 [并查集]

懒得找原题了，全是模板题。。。。

重量差异(b.cpp,1s,256MB)

【描述】
在实验室中，Nathan Wada作为助手的职责是测定两个样品的重量差异。当样品的差异很小时，使用天平比使用弹簧秤能得到更精确的结果，所以Nathan Wada只使用天平来测得一些样品的重量差。Nathan Wada偶尔会被询问一些样品的重量差，而他能否回答这些问题取决于在回答相应问题时他已经得到的测量结果。由于Nathan Wada所要处理的测量数据是巨大的，所以他希望你能写个程序帮他处理数据和回答问题。
【输入】
第一行包含两个整数 N 和 M ，其中 N 表示样品的数量，样品从 1 到 N 标号。
接下来 M 行，每行包括一个测量结果或者询问，按时间顺序给出。
一个测量结果被格式化为 ! a b w ，表示第 a 个样品比第 b 个样品轻 w 个单位重量，且任意的测试结果互不矛盾。
一个询问被格式化为 ? a b ，表示询问第 a 个样品比第 b 个样品轻多少个单位重量。
【输出】
对于每个询问输出一行，如果能回答问题，则输出问题的答案，你可以认为答案的绝对值不超过 1000000 。否则输出 UNKNOWN ，表示不能回答问题。
【样例输入】
4 7
! 1 2 100
? 2 3
! 2 3 100
? 2 3
? 1 3
! 4 3 150
? 4 1
【样例输出】
UNKNOWN
100
200
-50
【限制与约定】
对于30%的数据，N,M<=1000。
对于所有数据，1<=N,M<=100000，1<=a,b<=N。

思考

一开始以为要按秩合并，但是想一想就知道这只是一个裸的带权并查集而已。。什么树链剖分啊……等着多组数据TLE吧。。。
每个点存储它与根节点的重量差。

#include<cstdio>
#include<cassert>
#define maxn 100010
int fa[maxn],delt[maxn],n,m;
int findfa(int p)
{
    if(fa[p]!=p)
    {
        int cur=findfa(fa[p]);
        delt[p]+=delt[fa[p]];
        fa[p]=cur;
    }
    return fa[p];
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        if(n==0 && m==0) break;
        for(int i=1; i<=n; i++) fa[i]=i,delt[i]=0;
        while(m--)
        {
            char s[5]; scanf("%s",s);
            if(s[0]=='!')
            {
                int p,q,r;
                scanf("%d%d%d",&p,&q,&r);
                int p1=findfa(p),q1=findfa(q);
                if(p1==q1)
                {
                    assert(delt[p]-delt[q]==r);
                }
                else
                {
                    assert(delt[p1]==0);
                    delt[p1]=delt[q]+r-delt[p];
                    fa[p1]=q1;
                }
            }
            else
            {
                int p,q;
                scanf("%d%d",&p,&q);
                if(findfa(p)!=findfa(q)) puts("UNKNOWN");
                else printf("%d\n",delt[p]-delt[q]);
            }
        }
    }
    return 0;
}