[BZOJ4448][Scoi2015]情报传递（树上主席树）

题目：

我是超链接

题解：

大于c的有危险可以转化为小于等于i-c-1天发布任务，我们要想知道这些有多少人在这些天及之前发布任务，可以建立一棵权值线段树，每一个节点维护一段时间，节点的值维护个数。
那么对于查询一条路径的情况之前有过，只要给这棵树的节点建立根到自己的主席树，那么结果就是$sum(a)+sum(b)−sum(lca(a,b))−sum(fa(lca(a,b)))$

代码：

#include <cstdio>
#include <iostream>
#include <algorithm>
#define mc 18
using namespace std;
const int N=2e5+5;
struct hh{int l,r,w;}t[N*mc*5];
int tot,nxt[N*2],v[N*2],point[N],dfn[N],sz,nn,mi[mc+5],f[N][mc+5],h[N],jl[N];
int ss[N],L[N],R[N],id[N],K[N],root[N];
void addline(int x,int y)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
    ++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x;
}
void dfs(int x,int fa)
{
    dfn[++nn]=x;h[x]=h[fa]+1;
    for (int i=1;i<mc;i++)
      if (mi[i]>h[x]) break;
      else f[x][i]=f[f[x][i-1]][i-1];
    for (int i=point[x];i;i=nxt[i])
      if (v[i]!=fa) f[v[i]][0]=x,dfs(v[i],x);
}
int lca(int x,int y)
{
    if (h[x]<h[y]) swap(x,y);
    int k=h[x]-h[y];
    for (int i=0;i<mc;i++)
      if (k&(1<<i)) x=f[x][i];
    if (x==y) return x;
    for (int i=mc-1;i>=0;i--)
      if (f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    return f[x][0];
}
void insert(int &now,int l,int r,int x)
{
    t[++sz]=t[now]; now=sz;
    t[now].w++;
    if (l==r) return;
    int mid=(l+r)>>1;
    if (x<=mid) insert(t[now].l,l,mid,x);
    else insert(t[now].r,mid+1,r,x);
}
int qurry(int now,int l,int r,int lrange,int rrange)
{
    if (rrange<lrange) return 0;
    if (l>=lrange && r<=rrange) return t[now].w;
    int mid=(l+r)>>1,ans=0;
    if (lrange<=mid) ans+=qurry(t[now].l,l,mid,lrange,rrange);
    if (rrange>mid) ans+=qurry(t[now].r,mid+1,r,lrange,rrange);
    return ans;
}
int main()
{
    int n,x,y,k,gen,q;
    mi[0]=0;
    for (int i=1;i<mc;i++) mi[i]=mi[i-1]*2;
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        if (x==0) gen=i;
        else addline(i,x);
    }
    jl[gen]=1;
    dfs(gen,0);
    scanf("%d",&q);
    for (int i=1;i<=q;i++)
    {
        scanf("%d",&id[i]);
        if (id[i]==1) scanf("%d%d%d",&L[i],&R[i],&K[i]);
        else 
        {
            scanf("%d",&x);
            ss[x]=i;
        }
    }
    for (int i=1;i<=n;i++)
    {
        root[dfn[i]]=root[f[dfn[i]][0]];
        if (ss[dfn[i]]) insert(root[dfn[i]],1,q,ss[dfn[i]]);
    }
    for (int i=1;i<=q;i++)
      if (id[i]==1)
      {
        int fa=lca(L[i],R[i]),old=f[fa][0];
        int k=i-K[i]-1,ans4=0;
        int ans1=qurry(root[L[i]],1,q,1,k);
        int ans2=qurry(root[R[i]],1,q,1,k);
        int ans3=qurry(root[fa],1,q,1,k);
        if (old) ans4=qurry(root[old],1,q,1,k);
        printf("%d %d\n",h[L[i]]+h[R[i]]-h[fa]-h[old],ans1+ans2-ans3-ans4);
      }
}