• [dfs序][线段树][并查集] JZOJ P3766 大融合


    Description

    小强要在N个孤立的星球上建立起一套通信系统。这套通信系统就是连接N个点的一个树。这个树的边是一条一条添加上去的。在某个时刻,一条边的负载就是它所在的当前能够联通的树上路过它的简单路径的数量。
    这里写图片描述

    例如,在上图中,现在一共有了5条边。其中,(3,8)这条边的负载是6,因为有六条简单路径2-3-8,2-3-8-7,3-8,3-8-7,4-3-8,4-3-8-7路过了(3,8)。

    现在,你的任务就是随着边的添加,动态的回答小强对于某些边的负载的询问。

    Input

    第一行包含两个整数N,Q,表示星球的数量和操作的数量。星球从1开始编号。

    接下来的Q行,每行是如下两种格式之一:

    A x y 表示在x和y之间连一条边。保证之前x和y是不联通的。

    Q x y 表示询问(x,y)这条边上的负载。保证x和y之间有一条边。

    Output

    对每个查询操作,输出被查询的边的负载。

    Sample Input

    8 6

    A 2 3

    A 3 4

    A 3 8

    A 8 7

    A 6 5

    Q 3 8

    Sample Output

    6

    Data Constraint

    对于40%的数据,N,Q≤1000

    对于100%的数据,1≤N,Q≤100000

    题解

    按dfs序对每个点建权值线段树. 合并的时候直接合并两个点根的线段树
    查询的时候找到x,y所在树的根f
    假设deep[x]>deep[y] 
    那答案就是(size[f]-size[x])*size[x]; 
    (Tip:这里的size[x]是要从f的线段树里查的,因为合并的时候并没有更新x的线段树)
    

    代码

    #include<cstdio> 
    #include<iostream>
    #define N 100010
    using namespace std;
    int father[N],l[N],r[N],deep[N],visit[N],n,b,to[N],next[N<<1],cnt,num,nl[N*20],nr[N*20],root[N*20],size[N*20],q,x[N],y[N],k[N],f[N<<1];
    char ch[5];
    int getfather(int x){ if (father[x]!=x) father[x]=getfather(father[x]); return father[x];}
    void insert(int x,int y)
    {
        next[++num]=to[x];
        to[x]=num;
        f[num]=y;
    }
    void add(int &k,int l,int r,int x)
    {
        if (!k) k=++cnt;
        int mid=(l+r)/2;
        if (l==r)
        {
            size[k]=1;
            return;
        }
        if (x<=mid) add(nl[k],l,mid,x); else add(nr[k],mid+1,r,x);
        size[k]=size[nl[k]]+size[nr[k]];
    }
    int merge(int x,int y)
    {
        if (!x) return y;
        if (!y) return x;
        size[x]+=size[y];
        nl[x]=merge(nl[x],nl[y]);
        nr[x]=merge(nr[x],nr[y]);
        return x;
    }
    void dfs(int x,int fa)
    {
        l[x]=++b; 
        visit[x]=1; 
        add(root[x],1,n,b);
        for (int i=to[x];i;i=next[i])   
            if (f[i]!=fa)
            {
                deep[f[i]]=deep[x]+1;
                dfs(f[i],x);
            }
        r[x]=b;
    }
    int query(int k,int l,int r,int left,int right)
    {
        int mid=(l+r)/2;
        if (l==left&&r==right) return size[k];
        if (right<=mid) return query(nl[k],l,mid,left,right);
        else if (mid<left) return query(nr[k],mid+1,r,left,right);
             else return query(nl[k],l,mid,left,mid)+query(nr[k],mid+1,r,mid+1,right);
    }
    int main()
    {
        scanf("%d%d",&n,&q);
        for (int i=1;i<=q;i++)
        {
            scanf("%s",ch);
            scanf("%d%d",&x[i],&y[i]);
            if (ch[0]=='A')
            {
                k[i]=1;
                insert(x[i],y[i]);
                insert(y[i],x[i]);
            }
            else k[i]=2;
        }
        for (int i=1;i<=n;i++) if (!visit[i]) dfs(i,0);
        for (int i=1;i<=n;i++) father[i]=i;
        for (int i=1;i<=q;i++)
            if (k[i]==1)
            {
                int xx=getfather(x[i]),yy=getfather(y[i]);
                if (deep[xx]>deep[yy]) swap(xx,yy);
                root[xx]=merge(root[xx],root[yy]);
                father[yy]=xx;
            }
            else
            {
                int xx=x[i],yy=y[i];
                if (deep[xx]<deep[yy]) swap(xx,yy);
                int p=getfather(xx),v=query(root[p],1,n,l[xx],r[xx]);
                printf("%lld
    ",(long long)(size[root[p]]-v)*v);
            }
        return 0;
    }
  • 相关阅读:
    redis中的发布订阅(Pub/Sub)
    emmc基础技术8:操作模式3-interrupt mode
    Linux命令-tar
    git获取内核源码的方法
    Linux内核基础设施
    Linux内核简介
    emmc基础技术8:操作模式2-device identification mode
    内核子系统文档撰写方法
    eMMC基础技术6:eMMC data读写
    eMMC基础技术10:寄存器介绍
  • 原文地址:https://www.cnblogs.com/Comfortable/p/8412224.html
Copyright © 2020-2023  润新知