多校第一场——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;}