hdu4605

Posted 2021-01-17 zsben991126

两颗线段树，分别维护向左走向右走的情况

线段树的结点维护区间有多少点被路径经过了

离线读入所有询问，dfs遍历树的每一个结点，访问到v时解决对v的所有查询，在dfs过程中只需要维护根节点到v的链，线段树查询链上有多少结点值大于或小于询问的x，即能求出到达x的概率

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<stdio.h>
#include<iostream>
#include<stdlib.h>
#include<string.h>
/*
类似二叉搜索树
给定一颗n个结点的二叉树
给定数值x，从root开始沿着二叉树往下搜索，分三种情况
1.搜到叶子结点或者结点权值等于X，停止搜索
2.搜到结点权值w<x,搜索左右儿子几率各为1/2
3.w>=x,搜索左右儿子几率各为1/8，7/8
问v点被搜索到的概率是多少

两棵线段树保存向左走向右走的结果分布情况
离线+dfs
*/
#include<algorithm>
#include<vector>
#include<map>
using namespace std;
#define maxn 110000
#define lmin 1
#define rmax len
#define lson l,(l+r)/2,rt<<1
#define rson (l+r)/2+1,r,rt<<1|1
#define root lmin,rmax,1
#define now l,r,rt
#define int_now int l,int r,int rt
#define INF 999999999999
#define LL long long
#define mod 10007
struct list
{
    int father;
    int l,r;
    int w;
}node[maxn];//树的结点
vector<int>vec[maxn];//询问结点v的问题编号
struct listt
{
    int x,y;
    int v;
    int w;
}qq[maxn];//询问
map<int,int>mp;
int len;
int fs[maxn*2];//离散化后的权值
int st=0;
int sum[maxn*4][2];//两棵线段树
void updata(int ll,int rr,int x,int leap,int_now)
{
    if(ll>r||rr<l)return;
    if(ll<=l&&rr>=r)
    {
        sum[rt][leap]+=x;
        return;
    }
    updata(ll,rr,x,leap,lson);
    updata(ll,rr,x,leap,rson);
    sum[rt][leap]=sum[rt<<1][leap]+sum[rt<<1|1][leap];
}
int query(int ll,int rr,int leap,int_now)
{
    if(ll>r||rr<l)return 0;
    if(ll<=l&&rr>=r)return sum[rt][leap];
    return query(ll,rr,leap,lson)+query(ll,rr,leap,rson);
}
void dfs(int x)
{
    for(int i=0;i<vec[x].size();i++)//遍历结点x对应的每个询问
    {
        int bi=vec[x][i];
        int w=qq[bi].w;
        w=mp[w];
        int nm=query(w,w,0,root)+query(w,w,1,root);//nm只有在走不到x结点时才非0
        int ll=query(1,w-1,0,root);//root到x路径上结点权值小于w并且球向左走的点个数
        int lr=query(w+1,len,0,root);//root到x路径上结点权值大于w并且球向左走的点个数
        int rl=query(1,w-1,1,root);//root到x路径上结点权值小于w并且球向左走的点个数
        int rr=query(w+1,len,1,root);//root到x路径上结点权值小于w并且球向左走的点个数;
        if(nm!=0)continue;
        qq[bi].x=rl;
        qq[bi].y=(ll+rl)*3+lr+rr;
 
    }
    int id=mp[node[x].w];
    if(node[x].l!=0)
    {
        updata(id,id,1,0,root);//在向左走的线段树中更新当前结点的权值
        dfs(node[x].l);
        updata(id,id,-1,0,root);//回溯，取消本次更新
    }
    if(node[x].r!=0)
    {
        updata(id,id,1,1,root);
        dfs(node[x].r);
        updata(id,id,-1,1,root);
    }
}
int main()
{
    int T,n,m,u,l,r;
    scanf("%d",&T);
    while(T--)
    {
        st=0;
        memset(sum,0,sizeof(sum));
        memset(node,0,sizeof(node));
        mp.clear();
        scanf("%d",&n);
        len=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&node[i].w);
            fs[++st]=node[i].w;
            vec[i].clear();
        }
        scanf("%d",&m);
        while(m--)
        {
            scanf("%d%d%d",&u,&l,&r);
            node[l].father=node[r].father=u;
            node[u].l=l;
            node[u].r=r;
        }
        node[1].father=-1;
        int q,v,ww;
        scanf("%d",&q);
        for(int i=1;i<=q;i++)
        {
            scanf("%d%d",&v,&ww);
            fs[++st]=ww;
            qq[i].v=v;
            qq[i].w=ww;
            qq[i].x=qq[i].y=-1;
            vec[v].push_back(i);
        }
        sort(fs+1,fs+st+1);
        len=0;
        fs[0]=-1;
        for(int i=1;i<=st;i++)
        {
            if(fs[i]!=fs[i-1])
            {
                if(mp.find(fs[i])==mp.end())
                {
                    mp[fs[i]]=++len;
                }
            }
        }
        dfs(1);
        for(int i=1;i<=q;i++)
        {
            if(qq[i].x==-1)puts("0");
            else
            {
                printf("%d %d
",qq[i].x,qq[i].y);
            }
        }
    }
    return 0;
}