BZOJ4690 Never Wait for Weights(带权并查集模板题)

题意：

一共有n个物品；输入：？ a b w 表示b比a重w，！ a b 表示输出b比a重多少；

思路：

对每个？ a b w；建立a、b的并查集，维护dis数组。dis[x]表示x比x的父亲fa[x]重多少；

代码：

#include <bits/stdc++.h>

using namespace std;

int fa[100100];
long long dis[100100];
int n,m;

int fi(int x)
{
    if(fa[x]==x)
        return x;
    int root = fi(fa[x]);
    dis[x] += dis[fa[x]];
    fa[x] = root;
    return root;
}

void uni(int x,int y,int d)
{
    int fx = fi(x);
    int fy = fi(y);
    if(fx!=fy)
    {
        fa[fy] = fx;
        dis[fy] = dis[x]-dis[y]+d;
    }
}

int main()
{
    while(~scanf("%d%d",&n,&m)&&(n!=0||m!=0))
    {
        memset(dis,0,sizeof(dis));
        for(int i = 0;i<=n;i++)
            fa[i] = i;
        while(m--)
        {
            char s[5],op;
            int a,b,w;
            scanf("%s",s);
            op = s[0];
            if(op=='!')
            {
                scanf("%d%d%d",&a,&b,&w);
                uni(a,b,w);
            }
            else
            {
                scanf("%d%d",&a,&b);
                if (fi(a)!=fi(b))
                    printf("UNKNOWN\n");
                else
                    printf("%d\n",dis[b]-dis[a]);
            }
        }
    }
    return 0;
}