洛谷 P3899 [湖南集训] 谈笑风生 （主席树）

题意：

中文体面不写了

分析：

考虑两种情况，对于b点在a点上面的这种情况，直接计算就好。

主要考虑b点在a点下面的这种情况。

画个图可有看出来，答案就是（记a所在的高度是dega）dega到dega+k之间的所有点的size之和（size即这个点的子树的大小）

知道了这个后那么就可以转化为主席树进行计算。

区间k大的权值维护的是每个数，而这里改成维护节点高度，最后sum数组维护的就是size之和，那么查询就跟线段树一样了。

（代码写得有点丑，逃）

#include<bits/stdc++.h>

#define mm(a,b) memset(a,b,sizeof(a))
#define ACCELERATE (ios::sync_with_stdio(false),cin.tie(0))
#define pii pair<int,int>
#define pid pair<double,double>
#define pil pair<long long,long long>
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define rush() int T;scanf("%d",&T);while(T--)
#define sc(a) scanf("%d",&a)
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
#define PI acos(-1.0)
#define E exp(1.0)
//#define io
using namespace std;

const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3f;

//#define gc getchar
char buf[1<<21],*p1=buf,*p2=buf;
inline int gc(){return p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++;}
inline int read(){
    int ret=0,f=0;char c=gc();
    while(!isdigit(c)){if(c=='-')f=1;c=gc();}
    while(isdigit(c)){ret=ret*10+c-48;c=gc();}
    if(f)return -ret;return ret;
}

const int maxn=3e5+10;

ll Hash[maxn],a[maxn];
ll root[maxn*20],ls[maxn*20],rs[maxn*20],s[maxn*20];
ll sz,tot;

void insert(ll l,ll r,ll x,ll &y,ll val,ll k){
    y=++sz;
    s[y]=s[x]+val;
    if(l==r) return;
    ls[y]=ls[x],rs[y]=rs[x];
    ll mid=l+r>>1;
    if(k<=mid) insert(l,mid,ls[x],ls[y],val,k);
    else insert(mid+1,r,rs[x],rs[y],val,k);
}

ll query(ll L,ll R,ll l,ll r,ll x,ll y){
    if(x>y) return 0;
    if(r<x||l>y) return 0;
    if(x<=l&&r<=y)  {
        return s[R]-s[L];
    }
    ll mid=l+r>>1;
    if(y<=mid) return query(ls[L],ls[R],l,mid,x,y);
    else if(x>mid) return query(rs[L],rs[R],mid+1,r,x,y);
    else {
        return query(ls[L],ls[R],l,mid,x,mid)+query(rs[L],rs[R],mid+1,r,mid+1,y);
    }
}

struct node{
    ll to,next;
}edge[maxn<<1];

ll head[maxn];
ll dfn[maxn],size[maxn],dep[maxn],low[maxn],pos[maxn];
ll mx;

void addedge(ll u,ll v){
    edge[tot].to=v;
    edge[tot].next=head[u];
    head[u]=tot++;
}

void dfs(ll u,ll pre){
    dfn[u]=++tot;
    size[u]=1;
    pos[dfn[u]]=u;
    dep[u]=dep[pre]+1;
    mx=max(mx,dep[u]);
    for(ll i=head[u];i!=-1;i=edge[i].next){
        ll v=edge[i].to;
        if(v==pre) continue;
        dfs(v,u);
        size[u]+=size[v];
    }
    low[u]=tot;
}

int main(){
    #ifdef io
    freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);
    #endif

    tot=0;
    mm(head,-1);
    ll n,q;
    scanf("%lld%lld",&n,&q);
    rep(i,1,n-1){
        ll u,v;
        scanf("%lld%lld",&u,&v);
        addedge(u,v);
        addedge(v,u);
    }
    tot=0;
    dfs(1,1);
    for(ll i=1;i<=n;i++){
        insert(1,mx,root[i-1],root[i],size[pos[i]]-1,dep[pos[i]]);
    }
    while(q--){
        ll p,k;
        scanf("%lld%lld",&p,&k);
        printf("%lld\n",query(root[dfn[p]],root[low[p]],1,mx,dep[p]+1,min(dep[p]+k,mx-1))+min(dep[p]-1,k)*(size[p]-1));
    }
    return 0;
}