主席树维护dfs序——BZOJ3653/Luogu3899 谈笑风生

题面：BZOJ3653 Luogu3899
被luogu难度等级骗了。。。
首先看到子树的题就是dfs序了，我们用L[i]表示i的初访问戳，R[i]表示i的末访问戳
我们设size[i]表示子树大小（不包括i），deep[i]表示深度
首先我们可以发现a,b,c在一条链上，所以我们考虑这几种情况：

• b是a的祖先，根据乘法原理我们可以得到答案是size[a]∗min(deep[a]−1,k)
• a是b的祖先这种情况就是重点了，我们要求的就是∑L[a]<L[i]<=R[i]且0<deep[i]−deep[a]<=ksize[i]

针对第二种情况的求和，我们用权值线段树维护，然后可持久化就可以了。。。
而且主席树代码也不长啊。。。

#include <cstdio>#include <algorithm>#include <cmath>#include <cstring>#include <iostream>#include <ctime>#include <map>#include <queue>#include <cstdlib>#include <string>#include <climits>#include <set>#include <vector>using namespace std;typedef long long ll;int n,m,nedge=0,p[600010],nex[600010],head[600010];int deep[300010],fa[300010],l[300010],r[300010],cnt=0,s[300010];int rt[300010],ls[6000010],rs[6000010],sx[300010],tot=0;ll t[6000010];inline void addedge(int a,int b){    p[++nedge]=b;nex[nedge]=head[a];head[a]=nedge;}inline void dfs(int x,int dep){    deep[x]=dep;l[x]=++cnt;sx[cnt]=x;    for(int k=head[x];k;k=nex[k])if(p[k]!=fa[x]){        fa[p[k]]=x;dfs(p[k],dep+1);s[x]+=s[p[k]]+1;    }    r[x]=cnt;}inline void xg(int la,int &nod,int l,int r,int z,int w){    nod=++tot;t[nod]=t[la]+w;    if(l==r)return;    ls[nod]=ls[la];rs[nod]=rs[la];int mid=l+r>>1;    if(z<=mid)xg(ls[la],ls[nod],l,mid,z,w);    else xg(rs[la],rs[nod],mid+1,r,z,w);}inline ll ssum(int la,int nod,int l,int r,int i,int j){    if(l>=i&&r<=j)return t[nod]-t[la];    int mid=l+r>>1;ll ans=0;    if(i<=mid)ans+=ssum(ls[la],ls[nod],l,mid,i,j);    if(j>mid)ans+=ssum(rs[la],rs[nod],mid+1,r,i,j);    return ans;}int main(){    scanf("%d%d",&n,&m);    for(int i=1;i<n;i++){        int x,y;scanf("%d%d",&x,&y);        addedge(x,y);addedge(y,x);    }    dfs(1,1);    for(int i=1;i<=n;i++)xg(rt[i-1],rt[i],1,n,deep[sx[i]],s[sx[i]]);    for(int i=1;i<=m;i++){        int x,y;scanf("%d%d",&x,&y);        ll ans=(ll)s[x]*min(y,deep[x]-1);        if(deep[x]<n)ans+=ssum(rt[l[x]-1],rt[r[x]],1,n,deep[x]+1,min(n,deep[x]+y));        printf("%lld\n",ans);    }    return 0;}