首页 > 代码库 > 【BZOJ-3779】重组病毒 LinkCutTree + 线段树 + DFS序

【BZOJ-3779】重组病毒 LinkCutTree + 线段树 + DFS序

2024-08-29 00:01:44   219人阅读

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，这个用线段树很好实现。

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

Code

#include<iostream>#include<cstdio>#include<cstring>#include<cmath>#include<algorithm>using namespace std;#define LL long longinline 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 100010int 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\n",SgtTree::Query(x)); break;                }        }    return 0;}

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

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

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

【BZOJ-3779】重组病毒 LinkCutTree + 线段树 + DFS序


联系
我们