hdu 4605 Magic Ball Game

题意：有T组测试数据，N表示这棵树有N个点，下一行的N个数，表示每个点的权值W，M表示边的关系，接下来的M行，每行有三个数字，u,b,b，表示点a和点b分别是点u的左儿子和右儿子。Q表示查询数，接下来的Q行，每行两个数字，v和X。

        一个权值为X球从根节点开始下落，每落到一个节点的时候，1.如果X=W[i]，或者没有儿子节点了，球停止下落。2.如果X<W[i]，球各有1/2的概率落到左右儿子节点。3.如果X>W[i]，球有1/8的概率落到左儿子，有7/8的概率落到右儿子。问球落到点v的概率是多少（概率用7^x/2^y表示，即输出x和y就可以）？

        当知道点v的时候，我们要统计的是，从根节点到点v上有多少个节点权值是大于X的，有多少个节点权值是小于X的（并且分别是在左儿子还是右儿子），是否有节点的权值与X相等。用两棵树状数组维护（代表是在左儿子还是在右儿子），直接DFS一次，将DFS路径上的权值加入对应线状数组里，然后统计就可以了。

#include <iostream>#include <cstdio>#include <cstring>#include <vector>#include <map>#include <algorithm>using namespace std;const int N=1e5+5;#define LL(x) (x<<1)#define RR(x) (x<<1|1)#define MID(a,b) (a+((b-a)>>1))struct OP{    int w,idx;    OP(){}    OP(int w,int idx) : w(w),idx(idx) {}};struct BIT{    int T[N*2],LIM;    void init(int n)    {        LIM=n;        memset(T,0,sizeof(T));    }    int lowbit(int x){return x&(-x);}    void updata(int pos,int valu)    {        for(int i=pos;i<=LIM;i+=lowbit(i)) T[i]+=valu;    }    int query(int st,int ed)    {        int sum=0;        for(int i=ed;i>0;i-=lowbit(i)) sum+=T[i];        for(int i=st-1;i>0;i-=lowbit(i)) sum-=T[i];        return sum;    }}seg[2];int n,m,K;int W[N],len,adj[N][2];int ans1[N],ans2[N];map<int,int> H;vector<int> Y;vector<OP> op[N];void fun(int u){    for(int i=0;i<(int)op[u].size();i++)    {        int id=H[ op[u][i].w ],idx=op[u][i].idx;        ans1[idx]=ans2[idx]=0;        if(seg[0].query(id,id)||seg[1].query(id,id)) ans1[idx]=-1;        else        {            ans2[idx]+=seg[0].query(1,id)*3;            ans2[idx]+=seg[0].query(id,len);            int tmp=seg[1].query(1,id);            ans1[idx]+=tmp;            ans2[idx]+=tmp*3;            ans2[idx]+=seg[1].query(id,len);        }    }}void dfs(int u){    int id=H[W[u]];    for(int i=0;i<2;i++)    {        int v=adj[u][i];        if(v==-1) break;        seg[i].updata(id,1);        fun(v);        dfs(v);        seg[i].updata(id,-1);    }}int main(){    //freopen("06.in","r",stdin);    int t;    scanf("%d",&t);    while(t--)    {        scanf("%d",&n);        Y.clear(); H.clear();        for(int i=1;i<=n;i++)        {            op[i].clear();        adj[i][0]=adj[i][1]=-1;            scanf("%d",&W[i]);            Y.push_back(W[i]);        }        scanf("%d",&m);        for(int i=0;i<m;i++)        {            int u,a,b; scanf("%d%d%d",&u,&a,&b);            adj[u][0]=a; adj[u][1]=b;        }        scanf("%d",&K);        for(int i=0;i<K;i++)        {            int v,w; scanf("%d%d",&v,&w);            Y.push_back(w);            op[v].push_back(OP(w,i));        }        sort(Y.begin(),Y.end());        Y.erase( unique(Y.begin(),Y.end()),Y.end());        for(int i=0;i<(int)Y.size();i++) H[Y[i]]=i+1;        len=(int)Y.size();        seg[0].init(len+5); seg[1].init(len+5);        fun(1);        dfs(1);        for(int i=0;i<K;i++)        {            if(ans1[i]==-1) puts("0");            else printf("%d %d\n",ans1[i],ans2[i]);        }    }    return 0;}