bzoj 4448(LCA+主席树)(离线)

传送门
题解:先离线处理,以时间为下标维护区间内比i-c[i]小的数的个数,然后用LCA去查就好了。。。

晚睡了一个小时,原因就是因为ans4初值没有赋成0,我以前还一向以为自己初始化变量很严谨,事实上严谨个*!
引以为戒!!!
打破新纪录:非世界杯年/欧冠夜,最晚熬夜时间2017.9.17 2:03

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int MAXN=2e5+4;
int n,q,Root;
int head[MAXN],edge=0;
struct EDGE {
    int v,nxt;
}e[MAXN];
int f[20][MAXN],dis[MAXN];
int t[MAXN],x[MAXN],y[MAXN],c[MAXN],type[MAXN];
int sum[MAXN*20],lc[MAXN*20],rc[MAXN*20],root[MAXN],tim=0;
inline int read() {
    int x=0,f=1;char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}
    while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x;
}
inline void adde(int u,int v) {
    e[edge].nxt=head[u],e[edge].v=v,head[u]=edge++;
}
inline void da() {
    for (int j=1;(1<<j)<=n;++j)
        for (int i=1;i<=n;++i)
            f[j][i]=f[j-1][f[j-1][i]];
}
inline int lca(int x,int y) {
    if (dis[x]<dis[y]) x^=y^=x^=y;
    int t=dis[x]-dis[y];
    for (int i=0;i<=17;++i)
        if (t&(1<<i)) x=f[i][x];
    if (x==y) return x;
    for (int i=17;~i;--i)
        if (f[i][x]^f[i][y]) x=f[i][x],y=f[i][y];
    return f[0][x];
}
void insert(int pre,int &now,int l,int r,int x) {
    now=++tim,sum[now]=sum[pre]+1;
    if (l==r) return ;
    lc[now]=lc[pre],rc[now]=rc[pre];
    int mid=l+r>>1;
    if (x<=mid) insert(lc[pre],lc[now],l,mid,x);
    else insert(rc[pre],rc[now],mid+1,r,x);
}
int query(int pre,int rt,int l,int r,int L,int R) {
    if (L>R) return 0;//because of 0's existence
    if (L<=l&&r<=R) return sum[rt]-sum[pre];
    int mid=l+r>>1,ret=0;
    if (L<=mid) ret+=query(lc[pre],lc[rt],l,mid,L,R);
    if (mid<R) ret+=query(rc[pre],rc[rt],mid+1,r,L,R);
    return ret;
}
/*void print() {
    for (int i=1;i<=20;++i) printf("%d ",sum[i]);
    puts("");
}*/
void dfs(int p,int fa) {
    root[p]=root[fa];//上一层信息pushdown
    if (t[p]) insert(root[fa],root[p],1,q,t[p]);//print();
    for (int i=head[p];~i;i=e[i].nxt) {
        int v=e[i].v;
        dis[v]=dis[p]+1;
        f[0][v]=p;
        dfs(v,p);
    }
}
int main() {
//  freopen("bzoj 4448.in","r",stdin);
    root[0]=0;
    memset(t,0,sizeof(t));
    memset(head,-1,sizeof(head));
    n=read();
    for (register int i=1;i<=n;++i) {
        int pos=read();
        if (pos) adde(pos,i);
        else Root=i;
    }
    q=read();
    for (register int i=1;i<=q;++i) {
        type[i]=read();
        if (type[i]^1) {
            int pos=read();
            t[pos]=i;
        }
        else x[i]=read(),y[i]=read(),c[i]=read();
    }
    dis[Root]=1;
    dfs(Root,0);
    da();//init the LCA array
    for (register int i=1;i<=q;++i) {
        if (type[i]^2) {
            int fa=lca(x[i],y[i]),fe=f[0][fa];
            int ans=dis[x[i]]+dis[y[i]]-dis[fa]-dis[fe];
            printf("%d ",ans);
            int k=i-c[i]-1;
            int ans1=query(root[0],root[x[i]],1,q,1,k);
            int ans2=query(root[0],root[y[i]],1,q,1,k);
            int ans3=query(root[0],root[fa],1,q,1,k);
            int ans4=0;//what the bloody f**king d**n flipping ***!!!
            if (fa^Root) ans4=query(root[0],root[fe],1,q,1,k);
//          printf("%d %d %d %d\n",ans1,ans2,ans3,ans4);
            printf("%d\n",ans1+ans2-ans3-ans4);
        }
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值