hdu 4605：Magic Ball Game（主席树或者离线）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4605

题目大意：

一颗树，树上每个结点有0个或2个儿子结点。每个结点有一个权值。我们用一个带有权值的魔法球从根部抛下。对于每个结点，魔法球可能会向左或向右运动。

设魔法球的权值为X。

1.若X==w[u]，则魔法球会停在当前结点。

2.若x<w[u],1/2几率向左，1/2几率向右。

3.若x>w[u]，1/8几率向左，7/8几率向右。

然后对于每次查询，输入x，v  询问权值为x的魔法球到达结点v的概率是多少。

解题思路：

对于从根结点到结点v的路径，我们先判断路径中是否存在结点权值等于x的结点，如果存在，可能性为0.否则我们统计出路径中向左向右比x大和比它小的结点数目。就可以根据概率直接算出最后的答案。

Ac代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+5;
const int INF=1e9+7;
int dx[4]={0,1,0,-1};
int dy[4]={1,0,-1,0};
int n,m,q,cnt,root[maxn],dis[maxn];
vector<int> v;  //排序去重
int getid(int x)  { return lower_bound(v.begin(),v.end(),x)-v.begin()+1; }

struct node
{
    int root;
    int lson,rson,val;
}g[maxn];
struct Node
{
    int l,r;
    int sum[2];     //这里分别记录向左和向右的结点数目
}t[maxn*20];
void update(int l,int r,int &x,int y,int pos,int fg)    //更新点的数量  区分左右
{
    t[++cnt]=t[y],t[cnt].sum[fg]++,x=cnt;
    if(l==r)
        return ;
    int m=(l+r)/2;
    if(pos<=m)
        update(l,m,t[x].l,t[y].l,pos,fg);
    if(pos>m)
        update(m+1,r,t[x].r,t[y].r,pos,fg);
}
int query(int l,int r,int pl,int pr,int x,int y,int fg)     //查询路径上左右点的个数
{
    if(pl<=l&&r<=pr)
        return t[y].sum[fg]-t[x].sum[fg];
    int m=(l+r)/2,ans=0;
    if(pl<=m)
        ans+=query(l,m,pl,pr,t[x].l,t[y].l,fg);
    if(pr>m)
        ans+=query(m+1,r,pl,pr,t[x].r,t[y].r,fg);
    return ans;
}
void dfs(int u,int dep)     //深搜  更新点
{
    int l=g[u].lson,r=g[u].rson;
    dis[u]=dep;
    if(l!=-1)
    {
        update(1,n,root[l],root[u],getid(g[u].val),0);
        dfs(l,dep+1);
    }
    if(r!=-1)
    {
        update(1,n,root[r],root[u],getid(g[u].val),1);
        dfs(r,dep+1);
    }
}
int main()
{
    int QAQ;
    scanf("%d",&QAQ);
    while(QAQ--)
    {
        v.clear(),cnt=0;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            scanf("%d",&g[i].val),g[i].root=i,g[i].lson=-1,g[i].rson=-1,v.push_back(g[i].val);
        sort(v.begin(),v.end());    //离散化
        v.erase(unique(v.begin(),v.end()),v.end());
        scanf("%d",&m);
        for(int i=1;i<=m;i++)   //建图
        {
            int l,r,w;
            scanf("%d%d%d",&w,&l,&r);
            g[w].lson=l,g[w].rson=r;
        }
        dfs(1,0);   //遍历
        scanf("%d",&q);
        while(q--)
        {
            int vg,x;
            scanf("%d%d",&vg,&x);
            int px=getid(x);
            if(v[px-1]==x)  //如果存在结点权值为x的结点 判断是否在路径上
            {
                int t1=query(1,n,px,px,root[1],root[vg],0);
                int t2=query(1,n,px,px,root[1],root[vg],1);
                if(t1>0||t2>0)
                {
                    printf("0\n");  //输出为0
                    continue;
                }
            }
            if(v[px-1]>x)  px--;
            int lk=0,rk=0;
            if(px>=1)
            {
                lk=query(1,n,1,px,root[1],root[vg],0);  //找到左边小于x的点
                rk=query(1,n,1,px,root[1],root[vg],1);  //同上
            }
            printf("%d %d\n",rk,lk*3+rk*3+(dis[vg]-rk-lk));     //根据数量直接写出x y
        }
    }
    //system("pause");
}