BZOJ.4771.七彩树(可持久化线段树)

BZOJ

---

考虑没有深度限制，对整棵子树询问怎么做。
对于同种颜色中DFS序相邻的两个点\(u,v\)，在\(dfn[u],dfn[v]\)处分别\(+1\)，\(dfn[LCA(u,v)]\)处\(-1\)，这样答案就是求子树和了（同种颜色多余贡献的会被减掉）。
对于深度的限制，考虑维护\(\max\{dep\}\)棵线段树\(T_i\)，分别表示只考虑深度在\(1\sim i\)之间的点的贡献（下标依旧是DFS序）。因为我们发现，对于询问\((x,k)\)，求\(T_{dep[x]+k}\)这棵线段树中（也就是只考虑深度在\(1\sim dep[x]+k\)的点），区间\([dfn[x],ed\_dfn[x]]\)的和，就是答案。因为可能的问题是会统计到区间中深度\(<dep[x]\)的点，但是区间\([dfn[x],ed\_dfn[x]]\)中没有深度\(<dep[x]\)的点啊！
所以我们按深度维护出可持久化线段树就OK了。
对于深度由\(d-1\)变成\(d\)，考虑在\(T_d\)中深度为\(d\)的点\(x\)的贡献。
同样像最初提到的那样，找出与\(x\)颜色相同且DFS序相邻的两个点\(pre,nxt\)，把\(LCA(pre,nxt)\)的\(-1\)加回去，再减掉\(LCA(pre,x),LCA(x,nxt)\)的贡献即可（当然这两个LCA中有一个就是\(LCA(pre,nxt)\)）。对每种颜色用\(set\)维护一下前驱后继。
复杂度\(O(n\log n)\)。

注意下\(dep[x]+k\)对\(\max\{dep\}\)取\(\min\)。

感觉代码也没有那么难写啊

---

//57056kb   1440ms
#include <set>
#include <cstdio>
#include <cctype>
#include <vector>
#include <cstring>
#include <algorithm>
#define gc() getchar()
#define MAXIN 300000
//#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
typedef long long LL;
const int N=1e5+5;

int col[N],root[N],H[N],nxt[N],dep[N],fa[N],sz[N],son[N],top[N],Index,dfn[N],ref[N],R[N];
std::set<int> st[N];
std::vector<int> vec[N];
char IN[MAXIN],*SS=IN,*TT=IN;
struct Segment_Tree
{
    #define ls son[x][0]
    #define rs son[x][1]
    #define lson ls,l,m
    #define rson rs,m+1,r
    #define S N*37//2logn!
    int tot,sum[S],son[S][2];
    #undef S
    void Modify(int &x,int y,int l,int r,int p,int v)
    {
        x=++tot, sum[x]=sum[y]+v;
        if(l==r) {ls=rs=0; return;}
        int m=l+r>>1;
        p<=m ? (rs=son[y][1],Modify(ls,son[y][0],l,m,p,v)) : (ls=son[y][0],Modify(rs,son[y][1],m+1,r,p,v));
    }
    int Query(int x,int l,int r,int L,int R)
    {
        if(!x) return 0;
        if(L<=l && r<=R) return sum[x];
        int m=l+r>>1;
        if(L<=m)
            if(m<R) return Query(lson,L,R)+Query(rson,L,R);
            else return Query(lson,L,R);
        return Query(rson,L,R);
    }
}T;

inline int read()
{
    int now=0;register char c=gc();
    for(;!isdigit(c);c=gc());
    for(;isdigit(c);now=now*10+c-48,c=gc());
    return now;
}
inline void AE(int u,int v)
{
    nxt[v]=H[u], H[u]=v;
}
inline int LCA(int u,int v)
{
    if(u==-1||v==-1) return -1;
    while(top[u]!=top[v]) dep[top[u]]>dep[top[v]]?u=fa[top[u]]:v=fa[top[v]];
    return dep[u]>dep[v]?v:u;
}
void DFS1(int x)
{
    int mx=0; sz[x]=1, son[x]=0;//!
    for(int v=H[x]; v; v=nxt[v])
        dep[v]=dep[x]+1, DFS1(v), sz[x]+=sz[v], sz[v]>mx&&(mx=sz[v],son[x]=v);
}
void DFS2(int x,int tp)
{
    top[x]=tp, ref[dfn[x]=++Index]=x, vec[dep[x]].push_back(x);
    if(son[x])
    {
        DFS2(son[x],tp);
        for(int v=H[x]; v; v=nxt[v])
            if(v!=son[x]) DFS2(v,v);
    }
    R[x]=Index;
}
void Solve()
{
    int n=read(),m=read();
    for(int i=1; i<=n; ++i) col[i]=read();
    for(int i=2; i<=n; ++i) AE(fa[i]=read(),i);
    dep[1]=1, DFS1(1), DFS2(1,1);
    int mx=1;
    for(int i=1; i<=n; ++i) mx=std::max(mx,dep[i]);
    for(int i=1; i<=mx; ++i)
    {
        root[i]=root[i-1];
        for(int j=0,l=vec[i].size(); j<l; ++j)
        {
            int x=vec[i][j],c=col[x];
            if(!st[c].empty())
            {
                std::set<int>::iterator it=st[c].upper_bound(dfn[x]);
                int nxt=it==st[c].end()?-1:ref[*it];
                int pre=it==st[c].begin()?-1:ref[*(--it)];
                int r1=LCA(pre,nxt),r2=LCA(pre,x),r3=LCA(x,nxt);
                if(r1==r2 && r3!=-1) T.Modify(root[i],root[i],1,n,dfn[r3],-1);
                else if(r1==r3 && r2!=-1) T.Modify(root[i],root[i],1,n,dfn[r2],-1);
            }
            T.Modify(root[i],root[i],1,n,dfn[x],1), st[c].insert(dfn[x]);
        }
    }
    for(int i=1,ans=0; i<=m; ++i)
    {
        int x=read()^ans,k=read()^ans;
        printf("%d\n",ans=T.Query(root[std::min(mx,dep[x]+k)],1,n,dfn[x],R[x]));
    }
    T.tot=0, Index=0, memset(H,0,n+1<<2);
    for(int i=1; i<=n; ++i) std::set<int>().swap(st[i]);
    for(int i=1; i<=mx; ++i) std::vector<int>().swap(vec[i]);
}

int main()
{
    for(int T=read(); T--; Solve());
    return 0;
}

转载于:https://www.cnblogs.com/SovietPower/p/10441572.html