[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
,dfn
,sz,nn,mi[mc+5],f
[mc+5],h
,jl
;
int ss
,L
,R
,id
,K
,root
;
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);
}
}