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 100010const 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;}