BZOJ 4448 主席树+树链剖分(在线)

为什么题解都是离线的…… (抄都没法抄
搞一棵主席树
1 操作 新树上的当前节点设成1
2 操作 查max(i-xx-1,0)那棵树上这条路径上有多少个点是1
让你找经过了多少个点 查的时候用deep 搞出来就好了

//By SiriusRen
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 444444
#define mem(x,y) memset(x,y,sizeof(x))
int n,q,op,xx,yy,zz,first[N],v[N],next[N],tot,root,rt[N],ans;
int size[N],fa[N],son[N],deep[N],top[N],change[N],back[N],cnt;
struct Tree{int l,r,num;}tree[N*30];
void Add(int x,int y){v[tot]=y,next[tot]=first[x],first[x]=tot++;}
void add(int x,int y){Add(x,y),Add(y,x);}
void dfs(int x){
    size[x]=1;
    for(int i=first[x];~i;i=next[i])if(v[i]!=fa[x]){
        deep[v[i]]=deep[x]+1,fa[v[i]]=x,dfs(v[i]);
        size[x]+=size[v[i]];
        if(size[son[x]]<size[v[i]])son[x]=v[i];
    }
}
void dfs2(int x,int tp){
    change[x]=++cnt,back[cnt]=x;top[x]=tp;
    if(son[x])dfs2(son[x],tp);
    for(int i=first[x];~i;i=next[i])
        if(v[i]!=fa[x]&&v[i]!=son[x])dfs2(v[i],v[i]);
}
void push_up(int pos){tree[pos].num=tree[tree[pos].l].num+tree[tree[pos].r].num;}
int build(int l,int r){
    int pos=++cnt;
    if(l==r){return pos;}
    int mid=(l+r)>>1;
    tree[pos].l=build(l,mid),tree[pos].r=build(mid+1,r);
    return pos;
}
int insert(int o,int l,int r,int num){
    int pos=++cnt;tree[pos]=tree[o];
    if(l==r){tree[pos].num=1;return pos;}
    int mid=(l+r)>>1;
    if(mid>=num)tree[pos].l=insert(tree[o].l,l,mid,num);
    else tree[pos].r=insert(tree[o].r,mid+1,r,num);
    push_up(pos);return pos;
}
int query(int o,int l,int r,int L,int R){
    if(l>=L&&r<=R)return tree[o].num;
    int mid=(l+r)>>1; 
    if(mid<L)return query(tree[o].r,mid+1,r,L,R);
    else if(mid>=R)return query(tree[o].l,l,mid,L,R);
    else return query(tree[o].l,l,mid,L,R)+query(tree[o].r,mid+1,r,L,R);
}
int find(int x,int y,int wei){
    int fx=top[x],fy=top[y],temp=0;
    while(fx!=fy){
        if(deep[fx]<deep[fy])swap(x,y),swap(fx,fy);
        temp+=query(rt[wei],1,n,change[fx],change[x]),ans+=deep[x]-deep[fx]+1;
        x=fa[fx],fx=top[x];
    }
    if(deep[x]<deep[y])swap(x,y);
    ans+=deep[x]-deep[y]+1;
    return temp+query(rt[wei],1,n,change[y],change[x]);
}
int main(){
    mem(first,-1);scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&fa[i]);
        if(!fa[i])root=i;else add(i,fa[i]);
    }
    dfs(root),dfs2(root,root),cnt=0,rt[0]=build(1,n);
    scanf("%d",&q);
    for(int i=1;i<=q;i++){
        scanf("%d",&op);
        if(op==1){
            rt[i]=rt[i-1],ans=0;
            scanf("%d%d%d",&xx,&yy,&zz);
            zz=max(i-zz-1,0);
            printf("%d %d\n",ans,find(xx,yy,zz));
        }
        else scanf("%d",&xx),rt[i]=insert(rt[i-1],1,n,change[xx]);
    }
}

这里写图片描述

posted @ 2017-01-07 16:31  SiriusRen  阅读(229)  评论(0编辑  收藏  举报