D. Ralph And His Tour in Binary Country

传送门

http://codeforces.com/problemset/problem/894/D

给出一个平衡的二叉树，带边权，每组询问等价于求到A点距离不超过H的所有点到A点的距离和。

思考为什么是“平衡二叉树”，其具有层数不超过logn，只有每个节点最多有两个子的特点。
对每个点将其子树所有点到该点的距离维护成一个有序表，由于是静态的，排序即可。之后就可以使用lower_bound去找有多少个点满足条件，维护有序表的前缀和就可得解。
除了点A所在子树外，要沿着父节点向上爬，对路径上的每个节点及该节点的另一个子树求解，向上爬的过程中H是单调递减的。

思路是对的，结果维护有序表的时候没想清楚，用了动态开点的线段树，结果时间和空间都炸了。

下次出思路以后要想清楚需要维护的信息的本质，尝试用更简单的结构和方法去优化。

#include <cstdio>#include <cstring>#include <algorithm>#include <iostream>#include <queue>#include <vector>using namespace std;typedef long long lint;#define maxn (1000010)#define limit (10000000)int L[maxn], n, m;vector<lint> a[maxn], sum[maxn];struct node {    int v;    lint d;    node(int _v, lint _d) { v = _v; d = _d; }};queue<node> que;void build(int x){    while(!que.empty()) que.pop();    que.push(node(x, 0));    while(!que.empty()) {        node now = que.front(); que.pop();        a[x].push_back(now.d);        int y = now.v;        if((y<<1) <= n && now.d + L[y<<1] <= limit) {            que.push(node(y<<1, now.d + L[y<<1]));        }        if((y<<1|1) <= n && now.d + L[y<<1|1] <= limit) {            que.push(node(y<<1|1, now.d + L[y<<1|1]));        }    }    sort(a[x].begin(), a[x].end());    for(int i = 0, sz = a[x].size(); i < sz; i++) {        sum[x].push_back(a[x][i]);        if(i > 0) sum[x][i] += sum[x][i-1];    }}lint sol(int x, int k){    //printf("x = %d, k = %d\n", x, k);    lint ans = 0;    int p = lower_bound(a[x].begin(), a[x].end(), k) - a[x].begin();    if(p == a[x].size() || a[x][p] > k) p--;    if(p >= 0)        ans += (lint)k * (p + 1) - sum[x][p];    //printf("ans0 = %lld\n", ans);    while(x >> 1) {        k -= L[x];        if(k <= 0) break;        ans += k;        int y = x ^ 1;        p = lower_bound(a[y].begin(), a[y].end(), k - L[y]) - a[y].begin();        if(p == a[y].size() || a[y][p] > k - L[y]) p--;        if(p >= 0)            ans += (lint)(k - L[y]) * (p + 1) - sum[y][p];        //printf("ans = %lld\n", ans);        x >>= 1;    }    return ans;}int main(){    cin >> n >> m;    for(int i = 2; i <= n; i++) {        scanf("%d", &L[i]);    }    for(int i = 1; i <= n; i++) {        build(i);    }    for(int i = 0, A, H; i < m; i++) {        scanf("%d%d", &A, &H);        printf("%I64d\n", sol(A, H));    }    return 0;}