bzoj3653

主席树+dfs序

b在a上方时可以O(1)算出来，子树中就用主席树查询区间和，权值线段树的下标是深度，值是子树size-1，每次查询就行了。。。线段树合并挂了

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 3e5 + ;
vector<int> G[N];
int n, cnt, q, tot;
int mir[N], root[N * ], dep[N], lc[N * ], rc[N * ], in[N], out[N];
ll sum[N * ], size[N];
void update(int &x, int y, int l, int r, int p, ll d)
{
    x = ++cnt;
    sum[x] = sum[y] + d;
    if(l == r) return;
    int mid = (l + r) >> ;
    lc[x] = lc[y];
    rc[x] = rc[y];
    if(p <= mid) update(lc[x], lc[y], l, mid, p, d);
    else update(rc[x], rc[y], mid + , r, p, d);
}
ll query(int x, int y, int l, int r, int a, int b)
{
    if(l > b || r < a) return ;
    if(l >= a && r <= b) return sum[y] - sum[x];
    int mid = (l + r) >> ;
    return (query(lc[x], lc[y], l, mid, a, b) + query(rc[x], rc[y], mid + , r, a, b));
}
void dfs(int u, int last)
{
    size[u] = ;
    in[u] = ++tot;
    mir[in[u]] = u;
    for(int i = ; i < G[u].size(); ++i)
    {
        int v = G[u][i];
        if(v == last) continue;
        dep[v] = dep[u] + ;
        dfs(v, u);
        size[u] += size[v];
    }
    out[u] = tot;
}
int main()
{
    scanf("%d%d", &n, &q);
    for(int i = ; i < n; ++i)
    {
        int u, v;
        scanf("%d%d", &u, &v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    dep[] = ;
    dfs(, );
    for(int i = ; i <= n; ++i) update(root[i], root[i - ], , n, dep[mir[i]], size[mir[i]] - );
    while(q--)
    {
        int x, k;
        scanf("%d%d", &x, &k);
        ll ans = query(root[in[x] - ], root[out[x]], , n, dep[x] + , min(dep[x] + k, n)) + (ll)min(k, dep[x] - ) * (ll)(size[x] - );
        printf("%lld\n", ans);
    }
    return ;
}