hdu4605 Magic Ball Game,函数式线段树

hdu4605 Magic Ball Game

求某点到根的路径的blabla....

一看就是函数式数据结构。

把点的权值推到节点下面的边上。
离散化权值后，以权值建函数式线段树。
用函数式线段树，树保存两个值，路径边上连接左右儿子的边为该权值的分别有几个。

某条路径上的值就是两个节点对应函数式线段树的值相减。

离散化时加入一个0，一个inf权值，防止-1,+1越界

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define NN 100010

const int inf=1000001000;

int ch[NN][2];
int qu[NN][2];
int val[NN],b[NN*2];
int tn;

int tott,root[NN];

struct segtree{
    int ls,rs,l,r,w[2];
}t[NN*21];

int build(int l,int r){
    int now=++tott;
    t[now].l=l;t[now].r=r;t[now].w[0]=t[now].w[1]=0;
    if (l==r){
        t[now].ls=t[now].rs=0;return now;
    }
    int mid=(l+r)>>1;
    t[now].ls=build(l,mid);
    t[now].rs=build(mid+1,r);
    return now;
}

int insert(int pos,int pas,int func){
    int now=++tott;
    t[now]=t[pas];
    t[now].w[func]+=1;
    if (t[now].ls==t[now].rs) return now;
    int mid=(t[now].l+t[now].r)>>1;
    if (pos<=mid) t[now].ls=insert(pos,t[now].ls,func);
    else t[now].rs=insert(pos,t[now].rs,func);
    return now;
}

int queryp(int pos,int t1,int t2){
    if (t[t1].l==t[t1].r) return t[t2].w[0]-t[t1].w[0]+t[t2].w[1]-t[t1].w[1];
    int m=(t[t1].l+t[t1].r)>>1;
    if (pos<=m) return queryp(pos,t[t1].ls,t[t2].ls);
    else       return queryp(pos,t[t1].rs,t[t2].rs);

}

int queryseg(int l,int r,int t1,int t2,int func){
    if (t[t1].l==l&&t[t1].r==r) {
        return t[t2].w[func]-t[t1].w[func];
    }
    int m=(t[t1].l+t[t1].r)>>1;
    if (r<=m) return queryseg(l,r,t[t1].ls,t[t2].ls,func);
    else if (l>m) return queryseg(l,r,t[t1].rs,t[t2].rs,func);
    else{
      return queryseg(l,m,t[t1].ls,t[t2].ls,func)+\
             queryseg(m+1,r,t[t1].rs,t[t2].rs,func);
    }
}

void discret(int &tn,int n,int q){
    int i;
    for(i=1;i<=n;++i){
        b[i]=val[i];
    }
    for(i=1;i<=q;++i){
        b[++n]=qu[i][1];
    }
    b[n+1]=0;b[n+2]=inf;
    sort(b+1,b+n+3);
    tn=1;
    for(i=2;i<=n+2;++i){
        if (b[i]!=b[i-1]) b[++tn]=b[i];
    }
}

int bsch(int l,int r,int val){
    int m;
    while(l<=r){
        m=l+r>>1;
        if (b[m]==val) return m;
        if (val<b[m]) r=m-1;
        else l=m+1;
    }
}

void dfsbuild(int u){
    int vl=ch[u][0],vr=ch[u][1];
    int vv=bsch(1,tn,val[u]);
    if (vl) {root[vl]=insert(vv,root[u],0);dfsbuild(vl);}
    if (vr) {root[vr]=insert(vv,root[u],1);dfsbuild(vr);}
}

int main(){
    //freopen("4605in.txt","r",stdin);
    int n,i,aa,bb,cc,m,q;
    int tmp,tmp1,tmp2,tmp3,tmp4;
    int t1;
    int cas;
    scanf("%d",&cas);
    while(cas--){
        scanf("%d",&n);
        for(i=1;i<=n;++i){
            scanf("%d",&val[i]);
        }
        scanf("%d",&m);

        memset(ch,0,sizeof(ch));
        for(i=1;i<=m;++i){
            scanf("%d%d%d",&aa,&bb,&cc);
            ch[aa][0]=bb;ch[aa][1]=cc;
        }
        scanf("%d",&q);
        for(i=1;i<=q;++i){
            scanf("%d%d",&aa,&bb);
            qu[i][0]=aa;qu[i][1]=bb;
        }

        discret(tn,n,q);
        tott=0;
        root[1]=build(1,tn);
        dfsbuild(1);

        for(i=1;i<=q;++i){
            aa=qu[i][0];bb=qu[i][1];

            bb=bsch(1,tn,bb);
            t1=root[1];aa=root[aa];
            tmp=queryp(bb,t1,aa);
            tmp1=queryseg(1,bb-1,t1,aa,0);
            tmp2=queryseg(bb+1,tn,t1,aa,0);
            tmp3=queryseg(1,bb-1,t1,aa,1);
            tmp4=queryseg(bb+1,tn,t1,aa,1);
            if (tmp>0) printf("0\n");
            else  printf("%d %d\n",tmp3,tmp2+tmp4+3*(tmp1+tmp3));

        }
    }
    return 0;
}