bzoj4690: Never Wait for Weights（帶權並查集）

題目傳送門
。

解法：
我就寫過一次帶權並查集。
還好他不難想。。
看到這道題我就覺得是帶權並查集。。。

fa[i]表示祖先。
sum[i]表示跟祖先的差值。
xx表示x的祖先，yy表示y的祖先。
那麼每次x和y差w的時候
推式子？
x-xx=sum[x] – xx-x=-sum[x]
y-yy=sum[y]
y-x=w;

xx-yy?
將xx合併到yy。
xx-yy=sum[y]-sum[x]-w
那麼sum[xx]也就等於這個。

然後詢問的話判斷是否在同一個集合裡面。

程式碼實現：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
int sum[110000],fa[110000];
int findfa(int x) {
    if(fa[x]!=x) {
        int tt=fa[x];
        fa[x]=findfa(fa[x]);sum[x]+=sum[tt];
    }return fa[x];
}
int main() {
    int n,m;
    while(scanf("%d%d",&n,&m)&&n&&m) {
        for(int i=1;i<=n;i++)fa[i]=i;memset(sum,0,sizeof(sum));
        while(m--) {
            char s[5];int a,b,w;
            scanf("%s%d%d",s+1,&a,&b);
            if(s[1]=='!') {
                scanf("%d",&w);
                int xx=findfa(a),yy=findfa(b);
                if(xx!=yy) {fa[xx]=yy;sum[xx]=sum[b]-sum[a]-w;}
            }
            else {
                int xx=findfa(a),yy=findfa(b);
                if(xx!=yy)printf("UNKNOWN\n");
                else printf("%d\n",sum[b]-sum[a]);
            }
        }
    }
    return 0;
}