土豪的时代(带权并查集)

土豪的时代
Time Limit: 500 MS    Memory Limit: 32768 K
Total Submit: 459(84 users)    Total Accepted: 154(76 users)    Rating:     Special Judge: No
Description
土豪圈有一个习惯：从来不告诉别人自己到底有多少钱。但他们总是喜欢和其他土豪比较，来看看谁更土豪。于是每每几天，就会爆出一些关于土豪资产的消息，比如A土豪比B土豪多了3254万，C土豪比D土豪少2124万等等，从这些消息中也很容易推测出某两个土豪之间的资产关系。小破觉得和土豪交朋友是件非常愉快的事，但她想知道，某两个土豪，哪个更土豪。你能帮帮她么。
Input
多组测试数据
每组测试数据第一行有一个数n。( n<=100000 )
接下来有n行，每行包含一种操作。
操作分为两种：
1 a b m :表示a比b多m万元。
2 a b: 表示询问a比b多多少万元。
(0 < a,b < n)
(0 <= m <= 1000)

评测数据保证合法。

Output
对于每次的询问操作，输出其结果。当结果未知时，输出"?"。
Sample Input
5
1 1 2 10
2 1 3
1 3 4 10
1 2 3 10
2 4 1
Sample Output
?
-30
Source
哈尔滨理工大学第五届ACM程序设计竞赛（热身）

 

#include <iostream>

using namespace std;

constexpr int N=1e5+7;
int fa[N],d[N];
int getf(int x){
    if(x==fa[x])return x;
    int root= getf(fa[x]);
    d[x]+=d[fa[x]];
    return fa[x]=root;
}
void unite(int x,int y,int w){
    int root1= getf(x);
    int root2= getf(y);
    if(x==y) return;
    d[root1]=w-d[x]+d[y];
    fa[root1]=root2;
}
int main(){
    int n;
    while(cin>>n){
        for(int i=0;i<=n;i++){
            fa[i]=i;
            d[i]=0;
        }
        for(int i=1;i<=n;i++) {
            int op, a, b;
            scanf("%d%d%d", &op, &a, &b);
            if (op == 1) {
                int w;
                scanf("%d", &w);
                unite(a, b, w);
            } else {
                if (getf(a) != getf(b)) {
                    printf("?\n");
                }
                else{
                    printf("%d\n",d[a]-d[b]);
                }
            }
        }
    }
    return 0;
}