BZOJ 1095 [ZJOI2007]Hide 捉迷藏 ——动态点分治

【题目分析】

    这题好基啊。

    先把分治树搞出来。然后每个节点两个堆。

    第一个堆保存这个块里的所有点（即分治树中的所有儿子）到分治树上的父亲的距离。

    第二个堆保存分治树子树中所有儿子第一个堆的最大值。

    建一个答案堆，表示所有节点第二个堆的最大值和次大值的和。

    然后就是维护。

    两点间距离用LCA转RMQ

    调了4h，代码一度改到7k

【代码】

#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define maxe 200005
#define maxn 100005
#define inf (0x3f3f3f3f)
#define F(i,j,k) for (int i=j;i<=k;++i)
#define D(i,j,k) for (int i=j;i>=k;--i)
 
int h[maxe],to[maxe],ne[maxe],en=0,ban[maxn],n,m,rt,now,dst[maxn],Tree_rt,col[maxn],ans=-1,x,aim;
int b[maxn<<2],cnt=0,tag=0,pos[maxn],c[maxn<<2][20],_log[maxn<<2];
 
void add(int a,int b) {to[en]=b;ne[en]=h[a];h[a]=en++;}
struct Heap{
    priority_queue <int> heap,del;
    void Ins(int x){heap.push(x);}
    void Del(int x){del.push(x);}
    void Pop(){while (del.size()&&del.top()==heap.top()) del.pop(),heap.pop(); heap.pop();}
    int Top(){while (del.size()&&del.top()==heap.top()) del.pop(),heap.pop(); return heap.top();}
    int S_Top(){int tmp=Top(),ret;Pop();ret=Top();Ins(tmp);return ret;  }
    int Size(){return heap.size()-del.size();}
}s1[maxn],s2[maxn],lst;
 
int mx[maxn],siz[maxn],size,fa[maxn];
char opt[11];
 
void dfs(int o,int fa)
{
    if (!tag)b[++cnt]=o,pos[o]=cnt;
    siz[o]=1;mx[o]=0;
    for (int i=h[o];i>=0;i=ne[i])
        if (!ban[to[i]]&&to[i]!=fa)
        {
            dfs(to[i],o);
            if (!tag) b[++cnt]=o;
            siz[o]+=siz[to[i]];
            if (siz[to[i]]>mx[o]) mx[o]=siz[to[i]];  
        }
}
 
void dfs_rt(int o,int fa)
{
    if (max(mx[o],size-siz[o])<now) rt=o,now=max(mx[o],size-siz[o]);
    for (int i=h[o];i>=0;i=ne[i])
        if (!ban[to[i]]&&to[i]!=fa) dfs_rt(to[i],o);
}
 
int lca(int a,int b)
{
    int pa=pos[a],pb=pos[b];
    if (pa>pb) swap(pa,pb);
    int l=_log[pb-pa+1];
    return min(c[pa][l],c[pb-(1<<l)+1][l]);
}
 
int dist(int a,int b)
{return dst[a]+dst[b]-2*lca(a,b);}
 
void dfs_add(int o,int fa)
{
    s1[rt].Ins((dist(o,aim)));
    for (int i=h[o];i>=0;i=ne[i])
        if (!ban[to[i]]&&to[i]!=fa)
            dfs_add(to[i],o);
}
 
void Divide(int o,int fat)
{
    int root; 
    dfs(o,-1); size=siz[o]; now=inf;
    dfs_rt(o,-1); root=rt;
    if (fat) fa[root]=fat;
    ban[root]=1;
    aim=fat;
    dfs_add(root,-1);
    s2[fat].Ins(s1[root].Top());
    if (!col[root])
        s2[root].Ins(0);
    for (int i=h[root];i>=0;i=ne[i])
        if (!ban[to[i]]) Divide(to[i],root);
    if (s2[root].Size()>=2)
    {
        ans=max(ans,s2[root].Top()+s2[root].S_Top());
        lst.Ins(s2[root].Top()+s2[root].S_Top());
    }
}
 
void dfs_dist(int o,int fat)
{
    for (int i=h[o];i>=0;i=ne[i])
    if (to[i]!=fat)
        dst[to[i]]=dst[o]+1,dfs_dist(to[i],o);
}
 
void Delete(int o)
{
    if (s2[o].Size()>=2)
        lst.Del(s2[o].Top()+s2[o].S_Top());
    s2[o].Del(0);
    if (s2[o].Size()>=2)
    {
        ans=max(ans,s2[o].Top()+s2[o].S_Top());
        lst.Ins(s2[o].Top()+s2[o].S_Top());
    }
    if (s2[o].Size()>0) ans=max(ans,0);
    int now=o;
    while (now)
    {
        int fat=fa[now];
        if (s2[fat].Size()>=2)
            lst.Del(s2[fat].Top()+s2[fat].S_Top());
        int tmp=s1[now].Top();
        s1[now].Del(dist(o,fat));
        if (s1[now].Size()==0||tmp!=s1[now].Top())
        {
            s2[fat].Del(tmp);
            if (s1[now].Size()>0)
                s2[fat].Ins(s1[now].Top());
        }
        if (s2[fat].Size()>=2)
        {
            ans=max(ans,s2[fat].Top()+s2[fat].S_Top());
            lst.Ins(s2[fat].Top()+s2[fat].S_Top());
        }
        if (s2[fat].Size()>0) ans=max(ans,0);
        now=fat;
    }
}
 
void Insert(int o)
{
    if (s2[o].Size()>=2)
        lst.Del(s2[o].Top()+s2[o].S_Top());
    s2[o].Ins(0);
    if (s2[o].Size()>=2)
    {
        ans=max(ans,s2[o].Top()+s2[o].S_Top());
        lst.Ins(s2[o].Top()+s2[o].S_Top());
    }
    if (s2[o].Size()>0) ans=max(ans,0);
    int now=o;
    while (now)
    {
        int fat=fa[now];
        if (s2[fat].Size()>=2)
            lst.Del(s2[fat].Top()+s2[fat].S_Top());
        int tmp,flag=0;
        if (s1[now].Size()==0) flag=1;
        else tmp=s1[now].Top();
        s1[now].Ins(dist(o,fat));
        if (flag||tmp!=s1[now].Top())
        {
            if (!flag)
                s2[fat].Del(tmp);
            s2[fat].Ins(s1[now].Top());
        }
        if (s2[fat].Size()>=2)
        {
            lst.Ins(s2[fat].Top()+s2[fat].S_Top());
            ans=max(ans,s2[fat].Top()+s2[fat].S_Top());
        }
        if (s2[fat].Size()>0) ans=max(ans,0);
        now=fat;
    }
}
 
int main()
{
    memset(h,-1,sizeof h);
    scanf("%d",&n);
    F(i,1,n-1)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        add(a,b);add(b,a);
    }
    tag=1;
    dfs(1,-1);
    now=inf; size=siz[1]; dfs_rt(1,-1); dfs_dist(rt,-1);
    tag=0;
    dfs(rt,-1);
    tag=1;
    F(i,1,cnt) c[i][0]=dst[b[i]]; F(i,2,cnt) _log[i]=_log[i>>1]+1;
    F(j,1,_log[cnt])
        for (int i=1;i+(1<<j)-1<=cnt;++i)
            c[i][j]=min(c[i][j-1],c[i+(1<<j-1)][j-1]);
    Tree_rt=rt;
    Divide(1,0);
    scanf("%d",&m);
    F(i,1,m)
    {
        scanf("%s",opt);
        switch(opt[0])
        {
            case 'G': if (lst.Size()) ans=max(ans,lst.Top());printf("%d\n",ans); break;
            case 'C': scanf("%d",&x); ans=-1; if (!col[x]) Delete(x); else Insert(x); col[x]^=1; break;
        }
    }
}