bzoj 3779: 重组病毒【LCT+线段树维护dfs序】

%.8lf会WA!!%.8lf会WA!!%.8lf会WA!!要%.10lf!!
和4817有点像,但是更复杂。
首先对于操作一“在编号为x的计算机中植入病毒的一个新变种,在植入一个新变种时,病毒会在局域网中搜索核心计算机的位置,并沿着网络中最短的路径感染过去”,长得是不是有点像LCT中的access操作?进而发现,如果把同一颜色的点连起来作为LCT中的重边的话,那么询问二就相当于问路径上的虚边有多少。
假设没有换根操作,那么第二、三个操作是可以用树剖在线段树上维护的。 设每个点的权值val为这个点到根的路径上颜色个数,也就是虚边个数。那么考虑access操作的影响,对于他断开的重边,所在子树加一,对于他连上的重边,所在子树减一。直接在access过程中处理即可。
那么现在考虑换根操作:
首先出题人非常贴心的在换根操作后加了一个access,更加符合LCT的操作方式虽然不加应该就做不了了
首先对1为根进行dfs,把dfs序摊到线段树上,在这个线段树上维护修改和查询。
然后把平均数改成区间和,查完之后直接除即可(注意会爆int,注意精度问题)。
那么对于当先换根换到的root和要查询的子树x,用dfs序判断有三种情况:

  • \( root=x \),相当于求全区间和,直接返回即可;
  • x在root的子树中,那么换根对它没有影响,直接当做没有换根来查询即可;
  • x不在root的子树中,那么答案为全区间和减去x下面root所在的子树的区间和。
#include<iostream>
#include<cstdio>
using namespace std;
const int N=300005;
int n,m,h[N],cnt,fa[N],de[N],fr[N],ed[N],id[N],dfn,rl[N],root=1,s[N],top;
char c[10];
struct qwe
{
    int ne,to;
}e[N<<1];
struct xianduanshu
{
    int l,r;
    long long sum,lz;
}q[N<<2];
struct pinghengshu
{
    int f,c[2],tg;
}t[N];
int read()
{
    int r=0,f=1;
    char p=getchar();
    while(p>'9'||p<'0')
    {
        if(p==-1)
            f=-1;
        p=getchar();
    }
    while(p>='0'&&p<='9')
    {
        r=r*10+p-48;
        p=getchar();
    }
    return r*f;
}
void add(int u,int v)
{
    cnt++;
    e[cnt].ne=h[u];
    e[cnt].to=v;
    h[u]=cnt;
}
void build(int ro,int l,int r)
{
    q[ro].l=l,q[ro].r=r;
    if(l==r)
        return;
    int mid=(l+r)>>1;
    build(ro<<1,l,mid);
    build(ro<<1|1,mid+1,r);
}
void pd(int ro)
{
    q[ro<<1].sum+=q[ro].lz*(q[ro<<1].r-q[ro<<1].l+1);
    q[ro<<1].lz+=q[ro].lz;
    q[ro<<1|1].sum+=q[ro].lz*(q[ro<<1|1].r-q[ro<<1|1].l+1);
    q[ro<<1|1].lz+=q[ro].lz;
    q[ro].lz=0;
}
void update(int ro,int l,int r,int v)
{//cout<<l<<" "<<r<<" "<<v<<endl;
    if(l>r)
        return;
    if(q[ro].l==l&&q[ro].r==r)
    {
        q[ro].sum+=v*(q[ro].r-q[ro].l+1);
        q[ro].lz+=v;
        return;
    }
    if(q[ro].lz)
        pd(ro);
    int mid=(q[ro].l+q[ro].r)>>1;
    if(r<=mid)
        update(ro<<1,l,r,v);
    else if(l>mid)
        update(ro<<1|1,l,r,v);
    else
    {
        update(ro<<1,l,mid,v);
        update(ro<<1|1,mid+1,r,v);
    }
    q[ro].sum=q[ro<<1].sum+q[ro<<1|1].sum;
}
long long ques(int ro,int l,int r)
{
    if(l>r)
        return 0;
    if(q[ro].l==l&&q[ro].r==r)
        return q[ro].sum;
    if(q[ro].lz)
        pd(ro);
    int mid=(q[ro].l+q[ro].r)>>1;
    if(r<=mid)
        return ques(ro<<1,l,r);
    else if(l>mid)
        return ques(ro<<1|1,l,r);
    else
        return ques(ro<<1,l,mid)+ques(ro<<1|1,mid+1,r);
}
void dfs(int u,int fat)
{
    t[u].f=fa[u]=fat;
    de[u]=de[fat]+1;
    fr[u]=++dfn;
    update(1,fr[u],fr[u],de[u]);
    for(int i=h[u];i;i=e[i].ne)
        if(e[i].to!=fat)
            dfs(e[i].to,u);
    ed[u]=dfn;
}
bool srt(int x)
{
    return t[t[x].f].c[0]!=x&&t[t[x].f].c[1]!=x;
}
void pud(int x)
{
    if(t[x].tg)
    {
        t[x].tg=0;
        t[t[x].c[0]].tg^=1;
        t[t[x].c[1]].tg^=1;
        swap(t[x].c[0],t[x].c[1]);
    }
}
void zhuan(int x)
{
    int l,r,y=t[x].f,z=t[y].f;
    if(t[y].c[0]==x)
        l=0;
    else
        l=1;
    r=l^1;
    if(!srt(y))
    {
        if(t[z].c[0]==y)
            t[z].c[0]=x;
        else
            t[z].c[1]=x;
    }
    t[x].f=z;t[y].f=x;
    t[t[x].c[r]].f=y;
    t[y].c[l]=t[x].c[r];
    t[x].c[r]=y;
}
void splay(int x)
{//cout<<"splay"<<x<<endl;
    top=0;
    s[++top]=x;
    for(int i=x;!srt(i);i=t[i].f)
        s[++top]=t[i].f;
    while(top)
        pud(s[top--]);
    while(!srt(x))
    {
        int y=t[x].f,z=t[y].f;
        if(!srt(y))
        {
            if((t[y].c[0]==x)^(t[z].c[0]==y))
                zhuan(x);
            else
                zhuan(y);
        }
        zhuan(x);
    }
}
int zhao(int x,int y)
{
    for(int i=h[x];i;i=e[i].ne)
        if(e[i].to!=fa[x]&&fr[y]>=fr[e[i].to]&&ed[y]<=ed[e[i].to])
            return e[i].to;
    return 0;
}
void jia(int x,int w)
{
    if(x==root)
        update(1,1,n,w);
    else if(fr[root]>=fr[x]&&ed[root]<=ed[x])
    {
        int zi=zhao(x,root);
        update(1,1,fr[zi]-1,w);
        update(1,ed[zi]+1,n,w);
    }
    else
        update(1,fr[x],ed[x],w);
}
int wk(int x)
{
    pud(x);
    while(t[x].c[0])
        x=t[x].c[0],pud(x);
    return x;
}
void acc(int x)
{
    for(int i=0;x;i=x,x=t[x].f)
    {
        splay(x);
        if(t[x].c[1])
            jia(wk(t[x].c[1]),1);
        t[x].c[1]=i;
        if(t[x].c[1])
            jia(wk(t[x].c[1]),-1);
    }
}
void mkrt(int x)
{
    acc(x);
    splay(x);
    root=x;
    t[x].tg^=1;
}
double wen(int x)
{//cout<<ques(1,id[x],id[x]+si[x]-1)<<endl;
    if(x==root)
        return (double)ques(1,1,n)/n;
    if(fr[root]>=fr[x]&&ed[root]<=ed[x])
    {
        int zi=zhao(x,root);
        return (double)(ques(1,1,fr[zi]-1)+ques(1,ed[zi]+1,n))/(n-(ed[zi]-fr[zi]+1));
    }
    return (double)ques(1,fr[x],ed[x])/(ed[x]-fr[x]+1);
}
int main()
{
    n=read(),m=read();
    for(int i=1;i<n;i++)
    {
        int x=read(),y=read();
        add(x,y);add(y,x);
    }
    build(1,1,n);
    dfs(1,0);
    for(int i=2;i<=n;i++)
        t[i].f=fa[i];
    while(m--)
    {
        scanf("%s",c);
        if(c[2]=='L')
        {
            int x=read();
            acc(x);
        }
        else if(c[2]=='C')
        {
            int x=read();
            mkrt(x);
        }
        else
        {
            int x=read();
            printf("%.10lf\n",wen(x));
        }
    }
    return 0;
}
posted @ 2018-01-08 20:46  lokiii  阅读(177)  评论(0编辑  收藏  举报