HDU-4605-Magic Ball Game (离散化+树状数组+离线)

题目链接

题意：有一颗树，每个点有一个权值，从根节点放一个求往下落，球的值小于当前节点的值，往左右儿子落的概率各为1/2，相等则会停在这个节点，球的值大于当前节点的值往做儿子落的概率是1/8，往右儿子落的概率是7/8，q此询问，每个询问给一个点x和值w，问这个求落到x的概率。

思路：对于询问的每个点，只需要管跟节点到这个节点的这条链上的信息，用dfs可以很方便访问到每条链，把这条链分为向左和向右两部分，离散化所有权值后用树状数组维护每条链上的值的信息，对于每个节点的询问，只需要求出在这条链上值大于，等于，小于它的点各有多少个就好了，结果就很好算出来。

AC代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
vector<int> vec,edge[maxn];
int val[maxn],vis[maxn],fa[maxn],n,root;
int getid(int x) { return lower_bound(vec.begin(),vec.end(),x)-vec.begin()+1; }
struct node
{
    int v,id;
    node(){};
    node(int vv,int iid){v=vv;id=iid;};
}ans[maxn];
vector<node> Q[maxn];
int cnt[maxn],cnt2[maxn];
void add(int x,int val,int tp)
{
    if(tp==1){
        while(x<=n){
            cnt[x]+=val;
            x+=(x&-x);
        }
    }
    else{
        while(x<=n){
            cnt2[x]+=val;
            x+=(x&-x);
        }
    }
}
int sum(int x,int tp)
{
    int ans=0;
    if(tp==1){
        while(x>0){
            ans+=cnt[x];
            x-=(x&-x);
        }
    }
    else{
        while(x>0){
            ans+=cnt2[x];
            x-=(x&-x);
        }
    }
    return ans;
}
void dfs(int p)
{
    int num=Q[p].size();
    for(int i=0;i<num;i++){
        if(p==root){
            ans[Q[p][i].id]=node(0,0);
            continue;
        }
        int w=getid(Q[p][i].v);
        int a=sum(n,1),b=sum(w,1),c=sum(w-1,1);
        a-=b;b-=c;
        int a2=sum(n,2),b2=sum(w,2),c2=sum(w-1,2);
        a2-=b2;b2-=c2;
        if(b||b2) ans[Q[p][i].id]=node(-1,-1);
        else ans[Q[p][i].id]=node(c2,a+c*3+a2+c2*3);
    }
    if(edge[p].size()){
        int pos=getid(val[p]);
        add(pos,1,1);
        dfs(edge[p][0]);
        add(pos,-1,1);
        add(pos,1,2);
        dfs(edge[p][1]);
        add(pos,-1,2);
    }
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--){
        vec.clear();
        memset(vis,0,sizeof(vis));
        scanf("%d",&n);
        for(int i=1;i<=n;i++){
            scanf("%d",&val[i]);
            vec.push_back(val[i]);
            edge[i].clear();
            Q[i].clear();
        }
        int m,f,u,v;
        scanf("%d",&m);
        for(int i=1;i<=m;i++){
            scanf("%d%d%d",&f,&u,&v);
            edge[f].push_back(u);
            edge[f].push_back(v);
            vis[u]=1;
            vis[v]=1;
            fa[u]=fa[v]=f;
        }
        int q,x;
        scanf("%d",&q);
        for(int i=1;i<=q;i++){
            scanf("%d%d",&x,&v);
            vec.push_back(v);
            Q[x].push_back(node(v,i));
        }
        sort(vec.begin(),vec.end());
        vec.erase(unique(vec.begin(),vec.end()),vec.end());
        n=vec.size();
        for(int i=1;i<=n;i++) if(vis[i]==0){
            root=i;break;
        }
        dfs(root);
        for(int i=1;i<=q;i++){
            if(ans[i].v==-1) printf("0\n");
            else printf("%d %d\n",ans[i].v,ans[i].id);
        }
    }
    return 0;
}