hdu4605Magic Ball Game 树状数组

时间：2015-07-24 12:58:53 阅读：143 评论：0 收藏：0 [点我收藏+]

标签：

//给一棵树，树的每一个节点的子节点个数是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]) ;
}
}
}

标签：

原文地址：http://blog.csdn.net/cq_pf/article/details/47037897

踩

(0)

评论一句话评论（0）

分享档案

更多>

2021年07月29日 (22)
2021年07月28日 (40)
2021年07月27日 (32)
2021年07月26日 (79)
2021年07月23日 (29)
2021年07月22日 (30)
2021年07月21日 (42)
2021年07月20日 (16)
2021年07月19日 (90)
2021年07月16日 (35)

周排行