多校第一场——hdu4605——离线

题意：给出一棵二叉树，每个节点都有对应的权值。游戏规则是：从根节点开始向下“滚”一个权值为x的物体，到达一个节点的时候：

如果x=w[i]，即权值相等。则该物体停止移动。

如果x<w[i]，则该物体进入左儿子的概率是1/2，进入右儿子的概率是1/2。

如果x>w[i]，进入左儿子的概率是1/8，进入右儿子的概率是7/8。

询问是：一个权值为x的物体，能够到达u节点的概率是多少？

每个节点到根节点的路径都是唯一确定的，我们可以在一遍dfs中，处理掉所有的询问。

在dfs的过程中，用两个树状数组保存正在访问节点到根节点路径上所有节点的权值。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <cstring>
#include <vector>
#include <map>
using namespace std;

const int maxn = 300000 + 10;
vector<int> g[maxn];
vector<pair<int, int> > q[maxn];
map<int, int> mm;
int n, m, w[maxn], ind[maxn], root;
int c[2][maxn];
pair<int, int> ans[maxn];
inline int lowbit(int t) { return t & (-t); }
inline void add(int pos, int t, int cas) { while(pos < maxn) { c[cas][pos] += t; pos += lowbit(pos); } }
inline int sum(int pos, int cas) { int s = 0; while(pos > 0) { s += c[cas][pos]; pos -= lowbit(pos); } return s; }

void dfs(int u)
{
    for(int i = 0; i < q[u].size(); ++i)
    {
        pair<int, int> tmp = q[u][i];
        int ansx = 0, ansy = 0;
        if(sum(mm[tmp.first], 0) - sum(mm[tmp.first] - 1, 0) || sum(mm[tmp.first], 1) - sum(mm[tmp.first] - 1, 1))
            ans[tmp.second] = make_pair(-1, -1);
        else
        {
            ansx = sum(mm[tmp.first] - 1, 1);
            ansy += 3 * sum(mm[tmp.first] - 1, 1);
            ansy += sum(maxn - 1, 1) - sum(mm[tmp.first] - 1, 1);
            ansy += 3 * sum(mm[tmp.first] - 1, 0);
            ansy += sum(maxn - 1, 0) - sum(mm[tmp.first], 0);
            ans[tmp.second] = make_pair(ansx, ansy);
        }
    }
    for(int i = 0; i < g[u].size(); ++i)
    {
        int v = g[u][i];
        add(mm[w[u]], 1, i);
        dfs(v);
        add(mm[w[u]], -1, i);
    }
}

void prework()
{
    scanf("%d", &n); mm.clear();
    memset(c, 0, sizeof(c));
    for(int i = 1; i <= n; ++i)
    { 
        g[i].clear(); q[i].clear(); 
        scanf("%d", w + i);
        mm[w[i]] = 0;
        ind[i] = 0;
    }
    scanf("%d", &m);
    while(m--)
    {
        int u, a, b;
        scanf("%d %d %d", &u, &a, &b);
        g[u].push_back(a); ind[a]++;
        g[u].push_back(b); ind[b]++;
    }
    for(int i = 1; i <= n; ++i) if(ind[i] == 0) root = i;
    scanf("%d", &m);
    for(int i = 0; i < m; ++i)
    {
        int v, x;
        scanf("%d %d", &v, &x);
        q[v].push_back(make_pair(x, i));
        mm[x] = 0;
    }
    int tot = 0;
    for(map<int, int>::iterator it = mm.begin(); it != mm.end(); ++it) it->second = ++tot;
}

void solve()
{
    dfs(root);
    for(int i = 0; i < m; ++i)
    {
        pair<int, int> tmp = ans[i];
        if(tmp.first == -1 && tmp.second == -1) printf("0\n");
        else printf("%d %d\n", tmp.first, tmp.second);
    }
}

int main()
{
    int t; cin >> t;
    while(t--)
    {
        prework();
        solve();
    }
    return 0;
}