hdu4605Magic Ball Game 树状数组

//给一棵树,树的每一个节点的子节点个数是0或2
//对于每一个节点都有一个权值w[i]
//一个权值为x的球在每一个节点的情况有
//x=w[i] 这个球在该点不向下掉
//x<w[i] 这个球往左节点和右节点掉的概率各为1/2
//x>w[i] 这个球往左节点掉的概率为1/8 , 往右掉的概率为7/8
//问对于每一个权值为x其掉到节点为v的概率


//对于一颗树从一点到另一点的路径是确定的,只需要记录这条路径中
//往左走的路径中大于x的节点个数 l_grt,小于x的节点个数l_less
//往右走中大于x的节点个数 r_grt , 小于x的节点个数r_less
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std ;
const int maxn = 1e6+10 ;
int map[maxn][2] ;
vector<int> vec[maxn] ;
int x[maxn] , y[maxn] ;
int root[maxn] ;
int w_x[maxn] ;
int  w[maxn] ;
int a[maxn] ;int len ;
int tree[2][maxn] ;
void update(int x , int flag , int dx)
{
    while(x < maxn)
    {
        tree[flag][x] += dx ;
        x += x&(-x) ;
    }
}
int getsum(int x ,int flag)
{
    int sum = 0 ;
    while(x)
    {
        sum += tree[flag][x] ;
        x -= x&(-x) ;
    }
    return sum ;
}
void dfs(int u)
{
    //cout<<u<<endl;
    for(int i = 0;i < vec[u].size() ;i++)
    {
        int id = vec[u][i] ;
        int pos = lower_bound(a, a + len - 1, w_x[id]) - a + 1;
        int l_less = getsum(pos - 1 , 0) ;
        int r_less = getsum(pos - 1, 1) ;
        int l_all = getsum(len , 0) ;
        int r_all = getsum(len , 1) ;
        int l_grt = l_all - getsum(pos , 0) ;
        int r_grt = r_all - getsum(pos , 1) ;
        if(l_less + r_less + l_grt + r_grt != l_all + r_all)
        {
            x[id] = y[id] = -1 ;
            continue ;
        }
        x[id] = r_less ;
        y[id] = l_grt + r_grt + (l_less + r_less)*3 ;
    }
    for(int i = 0;i < 2;i++)
    {
        if(!map[u][i])continue ;
        int v = map[u][i] ;
        int pos = lower_bound(a , a + len - 1 , w[u]) - a + 1 ;
        update(pos , i , 1) ;
        dfs(v) ;
        update(pos , i , -1) ;
    }
}
int main()
{
    //freopen("in.txt" ,"r" , stdin) ;
    int T ;
    int N ;int M , Q ;
    scanf("%d" , &T) ;
    while(T--)
    {
        scanf("%d" , &N) ;
        len = 0 ;
        for(int i = 1;i <= N;i++)
        {
            scanf("%d" , &w[i]) ;
            a[len++] = w[i] ;
        }
        sort(a , a + N) ;
        len = unique(a , a + len) - a ;
        scanf("%d" ,&M) ;
        memset(map , 0 , sizeof(map)) ;
        memset(root , 0 , sizeof(root)) ;
        while(M--)
        {
            int u , l , r ;
            scanf("%d%d%d" , &u , &l , &r) ;
            map[u][0] = l ; map[u][1] = r ;
            root[l] = root[r] = 1;
        }
        scanf("%d" , &Q) ;
        for(int i = 1;i <= Q ;i++)
        {
            int v ;
            scanf("%d%d" , &v , &w_x[i]) ;
            vec[v].push_back(i) ;
        }
        int pos ;
        for(int i = 1;i <= N;i++)
        if(!root[i]){pos = i ;break;}
        dfs(pos) ;
        for(int i = 1;i <= Q;i++)
        {
            if(x[i]==-1)puts("0") ;
            else printf("%d %d\n"  , x[i] , y[i]) ;
        }
    }
}

































































你可能感兴趣的:(hdu4605Magic Ball Game 树状数组)