天天看点

[ZJOI2007]捉迷藏(动态点分治/(括号序列)(线段树))

题目描述

Jiajia和Wind是一对恩爱的夫妻,并且他们有很多孩子。某天,Jiajia、Wind和孩子们决定在家里玩捉迷藏游戏。他们的家很大且构造很奇特,由N个屋子和N-1条双向走廊组成,这N-1条走廊的分布使得任意两个屋子都互相可达。

游戏是这样进行的,孩子们负责躲藏,Jiajia负责找,而Wind负责操纵这N个屋子的灯。在起初的时候,所有的灯都没有被打开。每一次,孩子们只会躲藏在没有开灯的房间中,但是为了增加刺激性,孩子们会要求打开某个房间的电灯或者关闭某个房间的电灯。为了评估某一次游戏的复杂性,Jiajia希望知道可能的最远的两个孩子的距离(即最远的两个关灯房间的距离)。

我们将以如下形式定义每一种操作:

C(hange) i 改变第i个房间的照明状态,若原来打开,则关闭;若原来关闭,则打开。

G(ame) 开始一次游戏,查询最远的两个关灯房间的距离。

输入格式

第一行包含一个整数N,表示房间的个数,房间将被编号为1,2,3…N的整数。

接下来N-1行每行两个整数a, b,表示房间a与房间b之间有一条走廊相连。

接下来一行包含一个整数Q,表示操作次数。接着Q行,每行一个操作,如上文所示。

输出格式

对于每一个操作Game,输出一个非负整数到hide.out,表示最远的两个关灯房间的距离。若只有一个房间是关着灯的,输出0;若所有房间的灯都开着,输出-1。

这题本来老师是让用动态点分治做的,但是我用了一种括号序列的做法。

我们可以将一个点在dfs便历时放入一个左括号,回溯的时候放入一个右括号,这样就得到了一棵树的括号序列。

这个括号序列有一个性质,可以通过这个括号序列得到树上两点的距离。

例如

[ZJOI2007]捉迷藏(动态点分治/(括号序列)(线段树))

这棵树的括号序列是: ( 3 ( 5 ( 8 ) ) ( 4 ( 2 ) ( 6 ) ) ( 1 ( 7 ) ) ) (3(5(8))(4(2)(6))(1(7))) (3(5(8))(4(2)(6))(1(7)))

假如要求2到7的距离,可以截取2到7这一段括号序列 2 ) ( 6 ) ) ( 1 ( 7 2)(6))(1(7 2)(6))(1(7

删去数字和可以匹配的括号,剩下: ) ) ( ( ))(( ))((,一共四个括号,距离为4。

为什么有这个性质:

1 o 1^o 1o只添加了左括号

这样子的话,这个点肯定是u的祖先,在路径上

2 o 2^o 2o添加了左右括号

这样子的话,这个点肯定是祖先的其他儿子,或者是u点的儿子,产生不了贡献,所以要抵消。

3 o 3^o 3o左右括号都没添加

这样子的话,这个点肯定是祖先的其他儿子,产生不了贡献,不用计算。

4 o 4^o 4o只添加了右括号

这样子的话,有多少个右括号,就说明跳到lca需要多少步。

综述,上面的性质成立。

然后就到了维护答案的时候了。

设一个区间有 a a a个右括号, b b b个左括号,左儿子区间有 a 1 a1 a1个右括号, b 1 b1 b1个左括号,右儿子区间有 a 2 a2 a2个右括号, b 2 b2 b2个左括号。

那这个大区间的答案就是 a 1 + ∣ b 1 − a 2 ∣ + b 2 = m a x ( a 1 + b 1 − a 2 + b 2 , a 1 − b 1 + a 2 + b 2 ) = m a x ( ( a 1 + b 1 ) + ( b 2 − a 2 ) , ( a 1 − b 1 ) + ( a 2 + b 2 ) ) a1+|b1-a2|+b2=max(a1+b1-a2+b2,a1-b1+a2+b2)=max((a1+b1)+(b2-a2),(a1-b1)+(a2+b2)) a1+∣b1−a2∣+b2=max(a1+b1−a2+b2,a1−b1+a2+b2)=max((a1+b1)+(b2−a2),(a1−b1)+(a2+b2))。

显然 ( a 1 + b 1 ) , ( b 2 − a 2 ) , ( a 1 − b 1 ) , ( a 2 + b 2 ) (a1+b1),(b2-a2),(a1-b1),(a2+b2) (a1+b1),(b2−a2),(a1−b1),(a2+b2)都是可以区间单独维护的,前缀维护l1(a+b),l2(b-a),r1(a+b),r2(a-b)。

然后就是区间合并答案了。

1 o 1^o 1o(a,b)

{ a = a 1 , b = b 1 − a 2 + b 2 b 1 > a 2 a = a 1 + a 2 − b 1 , b = b 2 b 1 ⩽ a 2 \begin{cases}a=a1,b=b1-a2+b2&b1>a2\\a=a1+a2-b1,b=b2&b1\leqslant{a2}\end{cases} {a=a1,b=b1−a2+b2a=a1+a2−b1,b=b2​b1>a2b1⩽a2​

这个很好理解吧。

2 o 2^o 2o(l1,l2,b1,b2)

tree[hao].l1=max(tree[lc].l1,max(tree[rc].l1+tree[lc].a-tree[lc].b,tree[rc].l2+tree[lc].a+tree[lc].b));
tree[hao].l2=max(tree[lc].l2,tree[rc].l2-tree[lc].a+tree[lc].b);
tree[hao].r1=max(tree[rc].r1,max(tree[lc].r1-tree[rc].a+tree[rc].b,tree[lc].r2+tree[rc].a+tree[rc].b));
tree[hao].r2=max(tree[rc].r2,tree[lc].r2+tree[rc].a-tree[rc].b);
           

l1和l2是前缀的,所以整个区间要么是lson贡献的,要么是rson+lson贡献来的。

r1和r2同理。

3 o 3^o 3oans

直接按照推出来的式子计算即可

剩下的就是线段树基本操作了。

另:初始化时要 − I N F -INF −INF

#include<bits/stdc++.h>
#define lc hao<<1
#define rc hao<<1|1
#define inf 214748364
#define N 100010
using namespace std;
struct data
{
    int a,b,l1,l2,r1,r2,sum;//right left max(a+b) max(b-a) max(a+b) max(a-b) ans
}tree[N<<4];
int to[N<<1],nxt[N<<1],head[N],st[N<<2],nct,cnt,tot,val[N],n,m,x,y;
bool is[N];
char ch[2];
void adde(int x,int y)
{
    to[++nct]=y;
    nxt[nct]=head[x];
    head[x]=nct;
}
void dfs(int u,int fa)
{
    st[++cnt]=-1;
    st[++cnt]=u;
    val[u]=cnt;
    for(int i=head[u];i;i=nxt[i])
    {
        int v=to[i];
        if(v!=fa)
        {
            dfs(v,u);
        }
    }
    st[++cnt]=-2;
}
void up(int hao)
{
    if(tree[lc].b>tree[rc].a)
    {
        tree[hao].a=tree[lc].a;
        tree[hao].b=tree[lc].b-tree[rc].a+tree[rc].b;
    }else{
        tree[hao].b=tree[rc].b;
        tree[hao].a=tree[rc].a-tree[lc].b+tree[lc].a;
    }
    tree[hao].l1=max(tree[lc].l1,max(tree[rc].l1+tree[lc].a-tree[lc].b,tree[rc].l2+tree[lc].a+tree[lc].b));
    tree[hao].l2=max(tree[lc].l2,tree[rc].l2-tree[lc].a+tree[lc].b);
    tree[hao].r1=max(tree[rc].r1,max(tree[lc].r1-tree[rc].a+tree[rc].b,tree[lc].r2+tree[rc].a+tree[rc].b));
    tree[hao].r2=max(tree[rc].r2,tree[lc].r2+tree[rc].a-tree[rc].b);
    tree[hao].sum=max(max(tree[lc].sum,tree[rc].sum),max(tree[lc].r1+tree[rc].l2,tree[lc].r2+tree[rc].l1));
}
void change(int hao,int x)
{
    tree[hao].a=tree[hao].b=0;
    tree[hao].l1=tree[hao].r1=tree[hao].l2=tree[hao].r2=tree[hao].sum=-inf;
    if(st[x]==-1)
    {
        tree[hao].b=1;
    }else{
        if(st[x]==-2)
        {
            tree[hao].a=1;
        }else{
            if(!is[st[x]])
            {
                tree[hao].l1=tree[hao].r1=tree[hao].l2=tree[hao].r2=0;
            }
        }
    }
}
void build(int hao,int l,int r)
{
    if(l==r)
    {
        change(hao,l);
        return;
    }
    int mid=(l+r)>>1;
    build(lc,l,mid);
    build(rc,mid+1,r);
    up(hao);
}
void update(int hao,int l,int r,int x)
{
    if(l==r)
    {
        change(hao,l);
        return;
    }
    int mid=(l+r)>>1;
    if(x<=mid)
    {
        update(lc,l,mid,x);
    }else{
        update(rc,mid+1,r,x);
    }
    up(hao);
}
int main()
{
//  freopen("1.txt","r",stdin);
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        adde(x,y);
        adde(y,x);
    }
    dfs(1,-1);
    tot=n;
    build(1,1,cnt);
    scanf("%d",&m);
    while(m--)
    {
        scanf("%s",ch);
        if(ch[0]=='G')
        {
            if(tot==1)
            {
                puts("0");
            }else{
                if(tot==0)
                {
                    puts("-1");
                }else{
                    printf("%d\n",tree[1].sum);
                }
            }
        }else{
            scanf("%d",&x);
            if(is[x])
            {
                tot++;
                is[x]=0;
            }else{
                tot--;
                is[x]=1;
            }
            update(1,1,cnt,val[x]);
        }
    }
    return 0;
}