主席树维护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的祖先这种情况就是重点了，我们要求的就是$\sum_{L[a]<L[i]<=R[i]且0<deep[i]-deep[a]<=k} size[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;
}