主席樹維護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;
}