西科大第12届院赛 C 上决╇ф的旅程 【主席树】

传送门
题意: 给定一颗带权有根树, 即有边权也有点权, 每次询问一颗子树内与当前这颗子树的根的距离不超过L的点权和。

思路: 我们记录下每个点到根的距离, 设为dep[i], 那么询问相当于为de[to] - depu] <= L 的点劝和, 实际上就是dep[to] <= dep[u] + L, 其次还有就是通过dfs序可以把子树表示为一段区间, 所以问题就转化为了一段区间中小于等于某个值的和是多少, 那么很明显就是个经典主席树问题, 我们通过dep[i], 来建树, 然后对应的位置上面的权值为a[i], 这样每次查询的时候找一找小于等于dep[u] + L 的和是多少即可…这个问题和查找区间中小于等于某个数的和是多少一样的, 只不过这里的权值不是小于等于某个数的数的权值, 而是新给了一个权值, 那么加上新的权值而不是原先数的权值即可, 道理是一样的…….

AC Code(400ms+)

const int maxn = 1e5+5;
int n, m;
struct Tree {
    int ls, rs; ll val; // 左右儿子的编号, 和维护的一个值.
}tre[maxn*40];
int idx, root[maxn];
int build(int l, int r) {
    int nod = ++idx;
    tre[nod].val = 0;
    if (l == r) return nod;
    int mid = (l + r) >> 1;
    tre[nod].ls = build(l, mid);
    tre[nod].rs = build(mid+1, r);
    return nod;
}
int update(int pre, int l, int r, int pos, ll v) {
    int nod = ++idx;
    tre[nod] = tre[pre]; tre[nod].val += v;
    if (l == r) return nod;
    int mid = (l + r) >> 1;
    if (pos <= mid) tre[nod].ls = update(tre[pre].ls, l, mid, pos, v);
    else tre[nod].rs = update(tre[pre].rs, mid+1, r, pos, v);
    return nod;
}
ll query_sum(int ql, int qr, int l, int r, int pos) {
    if (l == r) return tre[qr].val - tre[ql].val;
    int mid = (l + r) >> 1;
    ll num = tre[tre[qr].ls].val - tre[tre[ql].ls].val;
    if (pos > mid) {
        return num + query_sum(tre[ql].rs, tre[qr].rs, mid+1, r, pos);
    }
    else return query_sum(tre[ql].ls, tre[qr].ls, l, mid, pos);
}
int p1[maxn], p2[maxn], tim;
int pos[maxn]; ll dep[maxn];
int cnt, head[maxn];
struct node {
    int to, next; ll w;
}e[maxn<<1];
void init() {
    cnt = tim = idx = 0;
    Fill(head, -1);
}
void add(int u, int v, ll w) {
    e[cnt] = node{v, head[u], w};
    head[u] = cnt++;
}
void dfs(int u, int fa) {
    p1[u] = ++tim;
    pos[tim] = u;
    for (int i = head[u] ; ~i ; i = e[i].next) {
        if (e[i].to == fa) continue;

        dep[e[i].to] = dep[u] + e[i].w;
        dfs(e[i].to, u);
    }
    p2[u] = tim;
}
ll a[maxn]; int rt;
vector<ll>ve;
int getid(ll x) {
    return upper_bound(ve.begin(), ve.end(), x) - ve.begin();
}
void solve() {
    scanf("%d%d", &n, &rt); init();
    for (int i = 1 ; i <= n ; i ++) scanf("%lld", a+i);
    for (int i = 1 ; i < n ; i ++) {
        int u, v; ll w;
        scanf("%d%d%lld", &u, &v, &w);
        add(u, v, w); add(v, u, w);
    }
    dfs(rt, -1);
    for (int i = 1 ; i <= n ; i ++) {
        ve.pb(dep[i]);
    }
    sort(ve.begin(), ve.end());
    ve.erase(unique(ve.begin(), ve.end()), ve.end());
    int len = sz(ve);
    root[0] = build(1, len);
    for (int i = 1 ; i <= n ; i ++) {
        root[i] = update(root[i-1], 1, len, getid(dep[pos[i]]), a[pos[i]]);
    }
    scanf("%d", &m);
    while(m--) {
        int u; ll L;
        scanf("%d%lld", &u, &L);
        int pos = getid(dep[u] + L);
        printf("%lld\n", query_sum(root[p1[u]-1], root[p2[u]], 1, len, pos));
    }
}

队友亲测: 暴力可过.(4000ms+)

猜你喜欢

转载自blog.csdn.net/Anxdada/article/details/81304108