题意:给出一棵二叉树,每个节点都有对应的权值。游戏规则是:从根节点开始向下“滚”一个权值为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;
}