zoukankan      html  css  js  c++  java
  • 【BZOJ-3779】重组病毒 LinkCutTree + 线段树 + DFS序

    3779: 重组病毒

    Time Limit: 20 Sec  Memory Limit: 512 MB
    Submit: 224  Solved: 95
    [Submit][Status][Discuss]

    Description

    黑客们通过对已有的病毒反编译,将许多不同的病毒重组,并重新编译出了新型的重组病毒。这种病毒的繁殖和变异能力极强。为了阻止这种病毒传播,某安全机构策划了一次实验,来研究这种病毒。
    实验在一个封闭的局域网内进行。局域网内有n台计算机,编号为1~n。一些计算机之间通过网线直接相连,形成树形的结构。局域网中有一台特殊的计算机,称之为核心计算机。根据一些初步的研究,研究员们拟定了一个一共m步的实验。实验开始之前,核心计算机的编号为1,每台计算机中都有病毒的一个变种,而且每台计算机中的变种都不相同。实验中的每一步会是下面中的一种操作:
    1、 RELEASE x
    在编号为x的计算机中植入病毒的一个新变种。这个变种在植入之前不存在于局域网中。
    2、 RECENTER x
    将核心计算机改为编号为x的计算机。但是这个操作会导致原来核心计算机中的病毒产生新变种,并感染过来。换言之,假设操作前的核心计算机编号为y,相当于在操作后附加了一次RELEASE y的操作。
    根据研究的结论,在植入一个新变种时,病毒会在局域网中搜索核心计算机的位置,并沿着网络中最短的路径感染过去。
    而第一轮实验揭露了一个惊人的真相:病毒的不同变种是互斥的。新变种在感染一台已经被旧变种感染的电脑时,会把旧变种完全销毁之后再感染。但研究员发现了实现过程中的漏洞。如果新变种在感染过程中尚未销毁过这类旧变种,需要先花费1单位时间分析旧变种,才能销毁。如果之前销毁过这类旧变种,就可以认为销毁不花费时间。病毒在两台计算机之间的传播亦可认为不花费时间。

    研究员对整个感染过程的耗时特别感兴趣,因为这是消灭病毒的最好时机。于是在m步实验之中,研究员有时还会做出如下的询问:
    3、 REQUEST x
    询问如果在编号为x的计算机的关键集合中的计算机中植入一个新变种,平均感染时间为多长。编号为y的计算机在编号为x的计算机的关键集合中,当且仅当从y沿网络中的最短路径感染到核心计算机必须经过x。由于有RECENTER操作的存在,这个集合并不一定是始终不变的。

    至此,安全机构认为已经不需要实际的实验了,于是他们拜托你编写一个程序,模拟实验的结果,并回答所有的询问。

    Input

    输入的第一行包含两个整数n和m,分别代表局域网中计算机的数量,以及操作和询问的总数。
    接下来n-1行,每行包含两个整数x和y,表示局域网中编号为x和y的计算机之间有网线直接相连。
    接下来m行,每行包含一个操作或者询问,格式如问题描述中所述。

    Output

    对于每个询问,输出一个实数,代表平均感染时间。输出与答案的绝对误差不超过 10^(-6)时才会被视为正确。

    Sample Input

    8 6
    1 2
    1 3
    2 8
    3 4
    3 5
    3 6
    4 7
    REQUEST 7
    RELEASE 3
    REQUEST 3
    RECENTER 5
    RELEASE 2
    REQUEST 1

    Sample Output

    4.0000000000
    2.0000000000
    1.3333333333

    HINT

    N < = 1 00 000 M < = 1 00 000 

    Source

    Solution

    这道题真是容易写残调半天。

    把题目转化一下发现这些过程实际上类似于LinkCutTree,具体的三个操作分别是:

    1、将x到根路径上的所有点染成一种新的颜色; 
    2、将x到根路径上的所有点染成一种新的颜色,并且把这个点设为根; 
    3、查询以x为根的子树中所有点到根 虚边数+1 的平均值。

    然后1对应Access,2对应makeroot,3维护子树和。

    LinkCutTree并不支持维护子树和,所以用dfs序+线段树维护,在每次Access的时候会对相应节点子树+1-1,这个用线段树很好实现。

    然后至于换根对线段树的影响,就一如既往的讨论三种情况即可。

    自己开始写的时候LCT直接link(x,y),一直有点小错误,然后发现,我的Link是无根树的写法,此处有向树,直接Link需要注意一下谁是谁的fa,然后就可以了,也可以直接利用DFS的信息赋值,然后再另其自生自灭。

    Code

    #include<iostream>
    #include<cstdio>
    #include<cstring>
    #include<cmath>
    #include<algorithm>
    using namespace std;
    #define LL long long
    inline int read()
    {
        int x=0,f=1; char ch=getchar();
        while (ch<'0' || ch>'9') {if (ch=='-') f=-1; ch=getchar();}
        while (ch>='0' && ch<='9') {x=x*10+ch-'0'; ch=getchar();}
        return x*f;
    }
    #define MAXN 100010
    int N,M;
    struct EdgeNode{int next,to,from;}edge[MAXN<<1];
    int head[MAXN],cnt=1;
    inline void AddEdge(int u,int v) {cnt++; edge[cnt].next=head[u]; head[u]=cnt; edge[cnt].to=v; edge[cnt].from=u;}
    inline void InsertEdge(int u,int v) {AddEdge(u,v); AddEdge(v,u);}
    int pl[MAXN],dfn,pre[MAXN],pr[MAXN],root,father[17][MAXN],deep[MAXN];
    LL val[MAXN];
    inline void DFS(int now,int last)
    {
        pl[now]=++dfn; pre[dfn]=now; val[now]=val[last]+1;
        for (int i=1; i<=16; i++)
            if (deep[now]>=(1<<i)) father[i][now]=father[i-1][father[i-1][now]];
                else break;
        for (int i=head[now]; i; i=edge[i].next)
            if (edge[i].to!=last)
                father[0][edge[i].to]=now,
                deep[edge[i].to]=deep[now]+1,
                DFS(edge[i].to,now);
        pr[now]=dfn;
    }
    inline int LCA(int x,int y)
    {
        if (deep[x]<deep[y]) swap(x,y);
        int dd=deep[x]-deep[y];
        for (int i=0; i<=16; i++) if (dd&(1<<i)) x=father[i][x];
        for (int i=16; i>=0; i--)
            if (father[i][x]!=father[i][y])
                x=father[i][x],y=father[i][y];
        return x==y? x:father[0][x];
    }
    inline int Jump(int x,int k)
    {
        for (int i=0; i<=16; i++)
            if (k&(1<<i)) x=father[i][x];
        return x;
    }
    namespace SgtTree
    {
        LL tree[MAXN<<2],tag[MAXN<<2]; int size[MAXN<<2];
        inline void Update(int now) {tree[now]=tree[now<<1]+tree[now<<1|1];}
        inline void Add(int now,LL val) {tree[now]+=(LL)size[now]*val; tag[now]+=val;}
        inline void PushDown(int now)
        {
            if (!tag[now] || size[now]==1) return;
            Add(now<<1,tag[now]); Add(now<<1|1,tag[now]);
            tag[now]=0;
        }
        inline void Build(int now,int l,int r)
        {
            size[now]=r-l+1;
            if (l==r) {tree[now]=val[pre[l]]; return;}
            int mid=(l+r)>>1;
            Build(now<<1,l,mid); Build(now<<1|1,mid+1,r);
            Update(now);
        }
        inline void Modify(int now,int l,int r,int L,int R,LL val)
        {
            if (L>R) return;
            PushDown(now);
            if (L<=l && R>=r) {Add(now,val); return;}
            int mid=(l+r)>>1;
            if (L<=mid) Modify(now<<1,l,mid,L,R,val);
            if (R>mid) Modify(now<<1|1,mid+1,r,L,R,val);
            Update(now);
        }
        inline LL QueryT(int now,int l,int r,int L,int R)
        {
            if (L>R) return 0;
            PushDown(now);
            if (L<=l && R>=r) return tree[now];
            int mid=(l+r)>>1; LL re=0;
            if (L<=mid) re+=QueryT(now<<1,l,mid,L,R);
            if (R>mid) re+=QueryT(now<<1|1,mid+1,r,L,R);
            return re;
        }
        inline int QueryS(int now,int l,int r,int L,int R)
        {
            if (L>R) return 0;
            PushDown(now);
            if (L<=l && R>=r) return size[now];
            int mid=(l+r)>>1,re=0;
            if (L<=mid) re+=QueryS(now<<1,l,mid,L,R);
            if (R>mid) re+=QueryS(now<<1|1,mid+1,r,L,R);
            return re;
        }
        inline double Query(int now)
        {
            if (!now) return 0;
            LL sum=0; int x,y,sz=0;
            if (now==root)
                sz=QueryS(1,1,N,pl[1],pr[1]),sum=QueryT(1,1,N,pl[1],pr[1]);
            else
                {
                    x=LCA(root,now);
                    if (x!=now)
                        sz=QueryS(1,1,N,pl[now],pr[now]),sum=QueryT(1,1,N,pl[now],pr[now]);
                    else
                        y=Jump(root,deep[root]-deep[now]-1),
                        sz=QueryS(1,1,N,1,pl[y]-1)+QueryS(1,1,N,pr[y]+1,N),
                        sum=QueryT(1,1,N,1,pl[y]-1)+QueryT(1,1,N,pr[y]+1,N);
                }
            return 1.0*sum/sz;
        }
        inline void Modify(int now,LL val)
        {
            if (!now) return;
            int x,y;
            if (now==root) Modify(1,1,N,pl[1],pr[1],val);
            else
                {
                    x=LCA(root,now);
                    if (x!=now)
                        Modify(1,1,N,pl[now],pr[now],val);
                     else 
                        y=Jump(root,deep[root]-deep[now]-1),
                        Modify(1,1,N,1,pl[y]-1,val),Modify(1,1,N,pr[y]+1,N,val);
                }
        }
    }using namespace SgtTree;
    namespace LCT
    {
        int fa[MAXN],son[MAXN][2],left[MAXN],right[MAXN]; bool rev[MAXN];
        inline bool is_root(int x) {return !fa[x] || son[fa[x]][0]!=x&&son[fa[x]][1]!=x;}
        inline void Pushup(int x)
        {
            if (!x) return;
            left[x]=right[x]=x;
            if (son[x][0]) left[x]=left[son[x][0]];
            if (son[x][1]) right[x]=right[son[x][1]];
        }
           inline void Rev(int x) {if (!x) return; swap(left[x],right[x]),swap(son[x][0],son[x][1]),rev[x]^=1;}
        inline void Pushdown(int x) {if (!x) return; if (rev[x]) Rev(son[x][0]),Rev(son[x][1]),rev[x]^=1;}
        inline void Rotate(int x)
        {
            int y=fa[x],w=son[y][1]==x,z=fa[y];
            son[y][w]=son[x][w^1];
            if (son[x][w^1]) fa[son[x][w^1]]=y;
            if (son[z][0]==y) son[z][0]=x; else if (son[z][1]==y) son[z][1]=x;
            fa[x]=z; fa[y]=x; son[x][w^1]=y; Pushup(y);
        }
        int stack[MAXN];
        inline void Splay(int x)
        {
            int t=x,top=0,y; stack[++top]=x;
            while (!is_root(t)) stack[++top]=t=fa[t];
            while (top) Pushdown(stack[top--]);
            while (!is_root(x))
                {
                    y=fa[x];
                    if (!is_root(y))
                        if ((son[fa[y]][0]==y)^(son[y][0]==x)) Rotate(x);
                            else Rotate(y);
                    Rotate(x);
                }
            Pushup(x);
        }
        inline void Access(int x)
        {
            for (int y=0; x; y=x,x=fa[x])
                Splay(x),
                SgtTree::Modify(left[y],-1),
                SgtTree::Modify(left[son[x][1]],1),
                son[x][1]=y,Pushup(x);
        }
        inline void Makeroot(int x) {Access(x); Splay(x); Rev(x);}
    }using namespace LCT;
    int main()
    {
        N=read(),M=read();
        for (int i=1,x,y; i<=N-1; i++) x=read(),y=read(),InsertEdge(x,y);
        DFS(root=1,0);
        SgtTree::Build(1,1,N);
        for (int i=1; i<=N; i++) LCT::fa[i]=father[0][i],LCT::left[i]=LCT::right[i]=i;
        while (M--)
            {
                char opt[10]; int x;
                scanf("%s",opt+1); x=read();
                switch (opt[3])
                    {
                        case 'L': LCT::Access(x); break;
                        case 'C': LCT::Makeroot(x); root=x; break;
                        case 'Q': printf("%.10lf
    ",SgtTree::Query(x)); break;
                    }
            }
        return 0;
    }

    突然想起char哥说这种题做的非常爽...然而,我因为线段树区间操作手贱达成了if (l==r)外加LCT中忘swap了一个地方两个脑残错误调了整整两节晚自习啊,感觉非常蛋疼;

    这说明以后写数据结构,思路一定要清晰,写的时候一定不能图快,要不然Debug浪费的时间更多。

    外加,自己常数怎么这么大啊,BZOJ成功倒数rank1......知乎

  • 相关阅读:
    『Nltk』常用方法
    『Kaggle』分类任务_决策树&集成模型&DataFrame向量化操作
    『Pandas』数据读取&DataFrame切片
    『TensotFlow』RNN中文文本_下_暨研究生开学感想
    『Scrapy』爬取斗鱼主播头像
    『Scrapy』爬取腾讯招聘网站
    『TensorFlow』梯度优化相关
    『Scrapy』终端调用&选择器方法
    『Scrapy』全流程爬虫demo
    使用VS2013、TFS2013和Git进行分布式团队协作
  • 原文地址:https://www.cnblogs.com/DaD3zZ-Beyonder/p/6216230.html
Copyright © 2011-2022 走看看