BZOJ 3653: 谈笑风生

题目在这里呀！
个人认为是一道很好的题目，原来可持久化线段树还能这么用，看题解之前还是没有想到啦要批评！那就写个题解补偿一下？

---

题意

给你一棵有根树，n个节点，有q次询问，每次询问，给出两个数x(1<=x<=n)，d，求有多少有序元组(y,z)满足 x,y,z互不相同，x,y均为z的祖先，且x,y之间的距离超过d。

题解

y的位置有两种情况
1、y是x的祖先
那么y一定是在x到根节点的路径上，有min(dep[x],d)种取法，然后z有size[x]种取法，故方案数为size[x]*min(dep[x],d)。
2、y是x的孩子
那么方案数为$\sum_{i}size[i]$（dep[i]-dep[x]<=d）
怎么求这个？考虑按dfs序来建可持久化线段树，然后深度来作为节点编号，节点里放size[]。吼吼吼为什么一开始没有想到没有想到？！
然后就迎刃而解啦（还傻傻地忘记写了一个return被嘲笑qwq）

//Suplex
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
#define ll long long
#define N 400400
using namespace std;
int n,Q,u,v,cnt,tot,vet[N+N],Next[N+N],head[N],L[N],R[N],opp[N],dep[N],size[N],p,k,maxd;
int root[N];

struct Segment_tree{
    int l,r;
    ll val;
}t[8000000];

void add_edge(int u,int v)
{
    vet[++cnt]=v;Next[cnt]=head[u];head[u]=cnt;
}

void dfs(int u,int fa)
{
    L[u]=++tot;opp[tot]=u;
    for(int i=head[u];i;i=Next[i]){
        int v=vet[i];
        if(v==fa) continue;
        dep[v]=dep[u]+1;
        dfs(v,u);
        size[u]+=size[v]+1;
    }
    R[u]=tot;
}

inline void modify(int &rt,int p,int l,int r,int x,int delta)
{
    rt=++cnt;
    t[rt].l=t[p].l;t[rt].r=t[p].r;
    t[rt].val=t[p].val+delta;
    if(l==r) return;
    int mid=(l+r)>>1;
    if(x<=mid) modify(t[rt].l,t[p].l,l,mid,x,delta);
    else modify(t[rt].r,t[p].r,mid+1,r,x,delta);
}

inline ll query(int rt,int l,int r,int x,int y)
{
    if(!rt) return 0;
    if(x<=l && r<=y) return t[rt].val;
    int mid=(l+r)>>1;
    if(y<=mid) return query(t[rt].l,l,mid,x,y);
    else if(x>mid) return query(t[rt].r,mid+1,r,x,y);
    else return query(t[rt].l,l,mid,x,mid)+query(t[rt].r,mid+1,r,mid+1,y);
}

int main()
{
    scanf("%d%d",&n,&Q);
    for(int i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        add_edge(u,v);add_edge(v,u);
    }
    dfs(1,-1);
    for(int i=1;i<=n;i++) maxd=max(maxd,dep[i]);
    cnt=0;
    for(int i=1;i<=n;i++) modify(root[i],root[i-1],0,maxd,dep[opp[i]],size[opp[i]]);
    while(Q--){
        scanf("%d%d",&p,&k);
        ll ans=1ll*size[p]*min(dep[p],k);
        ans+=query(root[R[p]],0,maxd,dep[p]+1,dep[p]+k);
        ans-=query(root[L[p]-1],0,maxd,dep[p]+1,dep[p]+k);
        printf("%lld\n",ans);
    }
    return 0;
}