传送门
题意: 给定一颗带权有根树, 即有边权也有点权, 每次询问一颗子树内与当前这颗子树的根的距离不超过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+)