带权并查集模板

题目:

题目描述
一个整数序列A{a0,a1,a2,....an-1},执行下列的操作:

  1. relate(x,y,z)表示ay比ax大z
  2. diff(x,y)打印输出ay-ax

输入

输出
打印操作输出值

样例输入
5 6
0 0 2 5
0 1 2 3
1 0 1
1 1 3
0 1 4 8
1 0 4

样例输出
2
?
10

code:

#include<cstdio>
using namespace std;
int fa[100005],val[100005];
int find(int x)
{
    if(fa[x]!=x)
    {
        int t=fa[x];
        fa[x]=find(fa[x]);
        val[x]+=val[t];
    }
    return fa[x];
}
int main()
{
    int n,q,tmp;
    scanf("%d%d",&n,&q);
    int pd,x,y,z;
    for(int i=0;i<n;i++)fa[i]=i;
    for(int i=1;i<=q;i++)
    {
        scanf("%d%d%d",&pd,&x,&y);
        if(pd==0)
        {
            scanf("%d",&z);
            int px=find(x);
            int py=find(y);
            if(px!=py)
            {
                fa[px]=py;
                val[px]=val[y]-val[x]+z;//注意这里的x,y和px之间的区别,不要弄混
            }
        }
        else
        if(pd==1)
        {
            if(find(x)!=find(y))
            {
                printf("?\n");
            }
            else
            printf("%d\n",val[x]-val[y]);
        }
    }
    return 0;
} 
全部评论

相关推荐

粗心的雪碧不放弃:纯学历问题,我这几个月也是一直优化自己的简历,后来发现优化到我自己都觉得牛逼的时候,发现面试数量也没有提升,真就纯学历问题
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务