[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 18using 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);      }}