重量差异 （加权并查集）

重量差异

10.11

思路：
加权并查集的经典应用。每个点存储它与根节点的重量差，合并与路径压缩时维护一下就好。
离线的按秩合并可能会T掉。。。

#include <cstdio>
#define LL long long
#define N 200010

int n, m;
int fa[N];
LL r[N];
char ss[210];

int getfather(int x){
    if(x == fa[x]) return x;
    int tmp = getfather(fa[x]);
    r[x] += r[fa[x]];//压缩时维护权值 
    return fa[x] = tmp;
}

int main(){
    freopen("b.in", "r", stdin);
    freopen("b.out", "w", stdout);
    scanf("%d%d", &n, &m);
    for(int i=1; i<=n; i++) fa[i] = i;
    while( m-- ){
        int u, v, w;
        scanf("%s", ss);
        if(ss[0] == '!'){
            scanf("%d%d%d", &u, &v, &w);
            int x = getfather(u), y = getfather(v);
            if(x != y)
                fa[x] = fa[y], r[x] = w + r[v] - r[u];
        }
        else{
            scanf("%d%d", &u, &v);
            int x = getfather(u), y = getfather(v);
            if(x != y) printf("UNKNOWN\n");
            else printf("%I64d\n", r[u]-r[v]);
        }
    }
    return 0;
}