hdu 4605 线段树与二叉树遍历

思路：

首先将所有的查询有一个vector保存起来。我们从1号点开始dfs这颗二叉树，用线段树记录到当前节点时，走左节点的有多少比要查询该节点的X值小的，有多少大的，

同样要记录走右节点的有多少比X小的，多少比X大的。小和大的x,y值题目给了。当要进行左儿子时，建该节点值插入走左的线段树，回退的时候将其删除，进入右儿子时将其插入走右的线段树，同样回退时删除。遍历完一个树，整个查询就做完了，最后输出。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#define Maxn 200100
#define lson(x) (x<<1)
#define rson(x) ((x<<1)+1)
#define mid ((tree[po].l+tree[po].r)>>1)
#define inf 100000000
using namespace std;
struct Query{
    int x,i;
};
struct AA{
    int x,y;
}ans[Maxn];
struct Tree{
    int l,r,lnum,rnum;
}tree[Maxn*];
int head[Maxn],e,w[Maxn],Index[Maxn],flag=,L,R,Lsum,Rsum;
vector<Query> query[Maxn];
struct Edge{
    int u,v,next;
}edge[Maxn];
void buildTree(int l,int r,int po)
{
    tree[po].l=l,tree[po].r=r;
    tree[po].lnum=,tree[po].rnum=;
    if(l==r)
        return ;
    buildTree(l,mid,lson(po));
    buildTree(mid+,r,rson(po));
}
void Insert(int val,int po,int type)
{
    if(tree[po].l==tree[po].r)
    {
        if(type)
            tree[po].rnum++;
        else
            tree[po].lnum++;
        return ;
    }
    if(type)
        tree[po].rnum++;
    else
        tree[po].lnum++;
    if(Index[mid]>=val)
        Insert(val,lson(po),type);
    else
        Insert(val,rson(po),type);
}
void del(int val,int po,int type)
{
    if(tree[po].l==tree[po].r)
    {
        if(type)
            tree[po].rnum--;
        else
            tree[po].lnum--;
        return ;
    }
    if(type)
        tree[po].rnum--;
    else
        tree[po].lnum--;
    if(Index[mid]>=val)
        del(val,lson(po),type);
    else
        del(val,rson(po),type);
}
void get_Num(int val,int po)
{
    if(Index[tree[po].l]>val)
        return ;
    if(Index[tree[po].r]<val)
    {
            R+=tree[po].rnum;
            L+=tree[po].lnum;
            return ;
    }
    if(tree[po].l==tree[po].r)
    {
        if(tree[po].rnum||tree[po].lnum)//如果图中有相同数值，则不能到达
            flag=;
        return ;
    }
    get_Num(val,lson(po));
    get_Num(val,rson(po));
}
void init()
{
    e=;
    Lsum=Rsum=;
    memset(head,-,sizeof(head));
    for(int i=;i<=;i++)
     {
         query[i].clear();
         ans[i].x=ans[i].y=;
     }
}
void add(int u,int v)
{
    edge[e].u=u;edge[e].v=v;edge[e].next=head[u];head[u]=e++;
}
void dfs(int u)//进行二叉树遍历
{
    int i,j,temp,Size,pos;
    Size=query[u].size();
    for(i=;i<Size;i++)
    {
        L=R=flag=;
        get_Num(query[u][i].x,);
        if(flag)
        {
            ans[query[u][i].i].y=-;
            continue;
        }
        ans[query[u][i].i].y+=L*;
        ans[query[u][i].i].y+=Lsum-L;
        ans[query[u][i].i].y+=R*;
        ans[query[u][i].i].y+=Rsum-R;
        ans[query[u][i].i].x+=R;
    }
    int f=;
    for(i=head[u];i!=-;i=edge[i].next)
    {
        if(f)
        {
            Insert(w[u],,);
            Rsum++;
        }
        else
        {
            Insert(w[u],,);
            Lsum++;
        }
        f++;
        dfs(edge[i].v);
        if(f==)
        {
            del(w[u],,);
            Rsum--;
        }
        else
        {
            del(w[u],,);
            Lsum--;
        }
    }
}
int main()
{
    int n,m,q,t,i,j,a,b,u,num;
    scanf("%d",&t);
    while(t--)
    {
        init();
        num=;
        scanf("%d",&n);
        for(i=;i<=n;i++)
         {
            scanf("%d",w+i);
            Index[i]=w[i];
         }
        sort(Index+,Index+n+);
        num=;
        for(i=;i<=n;i++)//将坐标离散化
        if(Index[i]>Index[num])
            Index[++num]=Index[i];
        buildTree(,num,);//建立线段树
        scanf("%d",&m);
        for(i=;i<=m;i++)
        {
            scanf("%d%d%d",&u,&a,&b);
            add(u,b);
            add(u,a);
        }
        scanf("%d",&q);
        Query temp;
        int v;
        for(i=;i<=q;i++)//将所有查询保存起来
        {
            scanf("%d%d",&v,&temp.x);
            temp.i=i;
            query[v].push_back(temp);
        }
        int f=;//标记走左还是右
        for(i=head[];i!=-;i=edge[i].next)//从1号节点开始遍历
        {
            if(f)//若走右
            {
                Insert(w[],,);//将该值插入右线段树，并且总数加1
                Rsum++;
            }
            else
            {
                Insert(w[],,);//插入左线段树,总数加1
                Lsum++;
            }
                f++;
            dfs(edge[i].v);
            if(f==)
            {
                del(w[],,);//从右儿子退出,将其删除，总数减1
                Rsum--;
            }
            else
            {
                del(w[],,);//从左儿子退出
                Lsum--;
            }
        }
        for(i=;i<=q;i++)
        {
            if(ans[i].y!=-)
                printf("%d %d\n",ans[i].x,ans[i].y);
            else
                printf("0\n");
        }
    }
    return ;
}