hdu 4605(树状数组+离散化)

题意：一棵树上每个节点都有权值，一个球从根节点往下走，每个球都有一个权值x，如果x大于当前节点的权值w[i]，有1/8的概率走左边孩子节点，有7/8的概率走右边孩子节点，如果x等于当前节点的权值w[i]或者当前节点是叶子节点，则球停止下落，如果x小于当前节点的权值，各1/2的概率走左右孩子节点。现在给出一个球的权值和它的目的节点，问能走到目的节点的概率，概率是7^x/2^y，输出x y。

题解：因为是一棵树，从根节点走到目的节点只有确定的一条路径可走，那么计算概率只需要统计，这条路径向左孩子节点走的节点的权值比x大的有ld个，比x小的有lx个，向右孩子节点走的节点的权值比x大的有rd个，比x小的有rx个，明显x = rd，y = (lx + rx) + (ld + rd) * 3。为了统计要走的路径中节点权值比x大的个数，可以树状数组，因为权值到1e9，要离散化处理。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int N = 100005;
int n, m, w[N], a[N << 2], cnt, res[N][2], C[2][N << 2];
vector<pair<int, int> > query[N];
vector<int> g[N];

int lowbit(int x) {
    return x & (-x);
}

int Sum(int x, int y) {
    int ret = 0;
    while (x > 0) {
        ret += C[y][x];
        x -= lowbit(x);
    }
    return ret;
}

void Add(int x, int d, int y) {
    while (x <= cnt) {
        C[y][x] += d;
        x += lowbit(x);
    }
}

void dfs(int u) {
    int sz = query[u].size();
    for (int i = 0; i < sz; i++) {
        int id = query[u][i].first;     
        int x = query[u][i].second; 
        int pos = lower_bound(a, a + cnt, x) - a + 1;
        int ld = Sum(pos - 1, 0);
        int rd = Sum(pos - 1, 1);
        int la = Sum(cnt, 0);
        int ra = Sum(cnt, 1);
        int lx = la - Sum(pos, 0);
        int rx = ra - Sum(pos, 1);
        if (ld + lx + rd + rx != la + ra) {
            res[id][0] = -1;
            continue;
        }
        res[id][0] = rd;
        res[id][1] = lx + rx + (ld + rd) * 3;
    }
    sz = g[u].size();
    for (int i = 0; i < sz; i++) {
        int v = g[u][i];
        int x = w[u];
        int pos = lower_bound(a, a + cnt, x) - a + 1;
        Add(pos, 1, i);
        dfs(v);
        Add(pos, -1, i);
    }
}

int main() {
    int t;
    scanf("%d", &t);
    while (t--) {
        scanf("%d", &n);
        cnt = 0;
        for (int i = 1; i <= n; i++) {
            scanf("%d", &w[i]);
            a[cnt++] = w[i];
            g[i].clear();
            query[i].clear();
        }
        scanf("%d", &m);
        int u, v1, v2, q, v, x;
        for (int i = 1; i <= m; i++) {
            scanf("%d%d%d", &u, &v1, &v2);
            g[u].push_back(v1);
            g[u].push_back(v2);
        }
        scanf("%d", &q);
        for (int i = 0; i < q; i++) {
            scanf("%d%d", &v, &x);
            query[v].push_back(make_pair(i, x));
            a[cnt++] = x;
        }
        sort(a, a + cnt);
        cnt = unique(a, a + cnt) - a;
        dfs(1);
        for (int i = 0; i < q; i++)
            if (res[i][0] != -1)
                printf("%d %d\n", res[i][0], res[i][1]);
            else
                printf("0\n");
    }
    return 0;
}