[RQNOJ36]数石子

题目大意：有N堆石子，每堆含石子个数未知。M个已知条件：[L，R]的区间内共有X个石子，K个问题：[L，R]的区间内共有多少石子？N≤5000 M，K≤10000

Click On 题解

与NOI《假面舞会》类似，用并查集的方式传递与父亲的关系

#include<cstdio>
int n,m,t,f[5005],d[5005];
int find(int i){
    if(f[i]==i) return i;
    int j=find(f[i]);
    d[i]+=d[f[i]];
    f[i]=j;
    return j;
}
int main(){
    int i,j,k,p,q;
    scanf("%d%d%d",&n,&m,&t);
    for(i=0;i<=n;i++) f[i]=i,d[i]=0;
    while(m--){
        scanf("%d%d%d",&i,&j,&k);
        p=find(i-1);
        q=find(j);
        if(p<q){
            f[q]=p;
            d[q]=k+d[i-1]-d[j];
        }
        else if(p>q){
            f[p]=q;
            d[p]=d[j]-d[i-1]-k;
        }
    }
    while(t--){
        scanf("%d%d",&i,&j);
        p=find(i-1);
        q=find(j);
        if(p!=q) printf("UNKNOWN\n");
        else printf("%d\n",(d[j]-d[i-1]));
    }
    return 0;
}