hdu 4605 Magic Ball Game
题意：给出一棵二叉树，树上每个节点都有一个权值w[i]，从根节点向下扔一个权值为x的小球，小球到达每个节点有三种情况：①x==w[u]，小球停止运动 ②x<w[u]，小球有相同的概率向左或向右走 ③x>w[u]，小球有1/8的概率向左走，有7/8的概率向右走。有q个询问，每个询问有两个值v、x，回答从根节点扔小球，小球停在v或经过v的概率。
思路：由于一棵树从根节点到任意一个节点的路径是唯一的，那么我们只要将询问离散化，然后用一个数据结构去维护走过的路径上的权值就行了，没走到一个结点，就回答和这个节点有关的询问。我用的线段树维护，因为小球向左向右走的概率跟该节点的权值和小球的权值有关，那么我们可以维护两颗线段树，一棵是从根节点到v的路径中向左走的节点的权值，一棵是从根节点到v的路径中向右走的节点的权值，然后就可以很简单的查到比小球权值大(小)的节点的个数，答案算起来也不难。走过某一节点后，就把这个节点从所选的数据结构中删掉就行了。当时多校做完以后看了题解没看懂Orz，今天也研究了好半天……但这题果断很好啊
代码：
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-9
#define pi acos(-1.0)
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
typedef long long ll;
const int maxn=100000+10;
int lson[maxn],rson[maxn],rnk[maxn];
int ans1[maxn],ans2[maxn],ww[maxn];
bool flag[maxn];
int n,m,q,cnt,lf,rn;
struct Node
{
int u,w;
bool operator < (const Node &a) const
{
return w<a.w;
}
};
Node node[maxn];
vector<Node>querys[maxn];
int maxv[maxn<<1][2],minv[maxn<<1][2],sum[maxn<<1][2];
void Init()
{
memset(lson,0xff,sizeof(lson));
memset(rson,0xff,sizeof(rson));
memset(flag,0,sizeof(flag));
for(int i=1;i<=n;++i) querys[i].clear();
}
void PushUp(int rt,int h)
{
sum[rt][h]=sum[rt<<1][h]+sum[rt<<1|1][h];
maxv[rt][h]=max(maxv[rt<<1][h],maxv[rt<<1|1][h]);
minv[rt][h]=min(minv[rt<<1][h],minv[rt<<1|1][h]);
}
void build(int l,int r,int rt)
{
sum[rt][0]=sum[rt][1]=0;
maxv[rt][0]=maxv[rt][1]=-1;
minv[rt][0]=minv[rt][1]=inf;
if(l==r) return ;
int m=(l+r)>>1;
build(l,m,rt<<1);
build(m+1,r,rt<<1|1);
}
int Query1(int l,int r,int rt,int v,int h)
{
if(maxv[rt][h]<v) return sum[rt][h];
if(minv[rt][h]>=v||l==r) return 0;
int m=(l+r)>>1;
int ls=rt<<1,rs=rt<<1|1;
if(maxv[ls][h]<v)
return sum[ls][h]+Query1(m+1,r,rs,v,h);
else return Query1(l,m,ls,v,h);
}
int Query2(int l,int r,int rt,int v,int h)
{
if(minv[rt][h]>v) return sum[rt][h];
if(maxv[rt][h]<=v||l==r) return 0;
int m=(l+r)>>1;
int ls=rt<<1,rs=rt<<1|1;
if(minv[rs][h]>v)
return sum[rs][h]+Query2(l,m,ls,v,h);
else return Query2(m+1,r,rs,v,h);
}
void Update(int p,int l,int r,int rt,int v,int h)
{
if(l==r)
{
if(v<0)
{
maxv[rt][h]=-1;minv[rt][h]=inf;
sum[rt][h]=0;
}
else
{
maxv[rt][h]=minv[rt][h]=v;
sum[rt][h]=1;
}
return;
}
int m=(l+r)>>1;
if(p<=m)
Update(p,l,m,rt<<1,v,h);
else Update(p,m+1,r,rt<<1|1,v,h);
PushUp(rt,h);
}
void dfs(int u)
{
int lbg,lsm,rbg,rsm;
int sz=querys[u].size();
Node tmp;
for(int i=0;i<sz;++i)
{
tmp=querys[u][i];
lsm=Query1(1,n,1,tmp.w,0);
rsm=Query1(1,n,1,tmp.w,1);
lbg=Query2(1,n,1,tmp.w,0);
rbg=Query2(1,n,1,tmp.w,1);
if(cnt-lsm-rsm-lbg-rbg>0)
ans1[tmp.u]=-1;
else
{
ans1[tmp.u]=rsm;
ans2[tmp.u]=lbg+rbg+(lsm+rsm)*3;
}
}
if(lson[u]<0) return ;
cnt++;
Update(rnk[u],1,n,1,ww[u],0);
dfs(lson[u]);
Update(rnk[u],1,n,1,-1,0);
Update(rnk[u],1,n,1,ww[u],1);
dfs(rson[u]);
Update(rnk[u],1,n,1,-1,1);
cnt--;
}
int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
int t;
scanf("%d",&t);
while(t--)
{
Init();
scanf("%d",&n);
for(int i=1;i<=n;++i)
{
scanf("%d",&ww[i]);
node[i].w=ww[i];
node[i].u=i;
}
sort(node+1,node+n+1);
for(int i=1;i<=n;++i)
rnk[node[i].u]=i;
scanf("%d",&m);
int u,a,b;
for(int i=0;i<m;++i)
{
scanf("%d%d%d",&u,&a,&b);
flag[a]=flag[b]=true;
lson[u]=a;rson[u]=b;
}
for(int i=1;i<=n;++i)
if(!flag[i]) {u=i;break;}
scanf("%d",&q);
Node tmp;
for(int i=0;i<q;++i)
{
scanf("%d%d",&a,&b);
tmp.u=i;tmp.w=b;
querys[a].push_back(tmp);
}
build(1,n,1);
lf=rn=cnt=0;
dfs(u);
for(int i=0;i<q;++i)
{
if(ans1[i]==-1)
printf("0\n");
else printf("%d %d\n",ans1[i],ans2[i]);
}
}
return 0;
}