牛客NOIP提高组R1 C保护(主席树)

题意

题目链接

sol

orz lyq

我们可以把一支军队(u, v)拆分为两个(u, lca)和(v, lca)

考虑一个点x，什么时候军队对它有贡献，肯定是u或v在他的子树内，且lca在他的子树外

因为需要让至少k个军队能够完全覆盖，所以肯定是选深度第k小的

这个过程可以用dfs序+主席树来实现

拿(u, lca)来说，在dfn[u]对应的线段树中，dep[lca]处+1即可。

然后查第k大即可

 

/**/
#include<cstdio>
#include<vector>
using namespace std;
const int maxn = 2 * 1e5 + 10;
inline int read() {
    char c = getchar(); int x = 0, f = 1;
    while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
    while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
    return x * f;
}
int n, m;
struct node {
    int siz;
}t[maxn * 70];
int root[maxn * 70], ls[maxn * 70], rs[maxn * 70];
int dep[maxn], fa[maxn], top[maxn], siz[maxn], son[maxn], deep[maxn], dfn[maxn], cnt, tot, num, id[maxn];
vector<int> v[maxn], q[maxn];//q[i] dfs搴忎负i鐨勯渶瑕佸姞鍏ョ殑鍏冪礌
void dfs1(int x, int _fa) {
    dfn[x] = ++cnt;
    fa[x] = _fa; siz[x]= 1; deep[x] = deep[_fa] + 1;
    for(int i = 0; i < (int)v[x].size(); i++) {
        int to = v[x][i];
        if(to == _fa) continue;
        dfs1(to, x);
        siz[x] += siz[to];
        if(siz[to] > siz[son[x]]) son[x] = to;
    }
}
void dfs2(int x, int topf) {
    top[x] = topf;
    if(!son[x]) return ;
    dfs2(son[x], topf);
    for(int i = 0; i < (int)v[x].size(); i++) {
        int to = v[x][i];
        if(top[to]) continue;
        dfs2(to, to);
    }
}
int lca(int x, int y) {
    while(top[x] != top[y]) {
        if(deep[top[x]] < deep[top[y]]) swap(x, y);
        x = fa[top[x]];
    }
    if(deep[x] > deep[y]) swap(x, y);
    return x;
}
void insert(int &k, int p, int l, int r, int pos) {
    if(l > r) return ;
    if(!k) k = ++tot, t[k].siz = t[p].siz + 1;
    if(l == r) return ;
    int mid = (l + r) >> 1;
    if(pos <= mid) rs[k] = rs[p], insert(ls[k], ls[p], l, mid, pos);
    else ls[k] = ls[p], insert(rs[k], rs[p], mid + 1, r, pos);
}
int query(int rl, int rr, int k, int l, int r) {
    if(t[rr].siz - t[rl].siz < k) return 0;
    if(l == r) return k > (t[rr].siz - t[rl].siz) ? 0 : l;
    int si = t[ls[rr]].siz - t[ls[rl]].siz;
    int mid = (l + r) >> 1;
    if(si >= k) return query(ls[rl], ls[rr], k, l, mid);
    else return query(rs[rl], rs[rr], k - si, mid + 1, r);
}
int main() {
  //  freopen("a.in", "r", stdin);
  //  freopen("b.out", "w", stdout);
    n = read(); m = read();
    for(int i = 1; i <= n - 1; i++) {
        int x = read(), y = read();
        v[x].push_back(y);
        v[y].push_back(x);
    }
    dfs1(1, 0);
    dfs2(1, 1);
    for(int i = 1; i <= m; i++) {
        int x = read(), y = read(), lca = lca(x, y);
        q[dfn[x]].push_back(deep[lca]);
        q[dfn[y]].push_back(deep[lca]);
    }
    for(int i = 1; i <= n; i++) {
        for(int j = 0; j < (int)q[i].size(); j++) {
            int x = q[i][j];
            ++num;
            insert(root[num], root[num - 1], 1, n, x);
        }
        id[i] = num;
    }
    int q = read();
    while(q--) {
        int u = read(), k = read();
        int ans = query(root[id[dfn[u] - 1]], root[id[dfn[u] + siz[u] - 1]], k, 1, n);
        if(ans == 0 || (deep[u] - ans <= 0)) printf("0\n");
        else printf("%d\n", deep[u] - ans);
    }
    return 0;
}
/**/