bzoj4551 [Tjoi2016&Heoi2016]树 树链剖分+树状数组+二分答案

Description

---

在2016年，佳媛姐姐刚刚学习了树，非常开心。现在他想解决这样一个问题：给定一颗有根树（根为1），有以下
两种操作：1. 标记操作：对某个结点打上标记（在最开始，只有结点1有标记，其他结点均无标记，而且对于某个
结点，可以打多次标记。）2. 询问操作：询问某个结点最近的一个打了标记的祖先（这个结点本身也算自己的祖
先）你能帮帮他吗?

1 ≤ N, Q ≤ 100000

Solution

---

握漕竟然1a，大概只打了50min的样子
一上来就想了在线做法，树链剖分然后每次二分一个祖先求路径上标记的和
这里只存在单点修改区间查询，那么树状数组即可
看了一波别人的题解发现原来可以离线做，我还是太弱了

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define drp(i,st,ed) for (int i=st;i>=ed;--i)
#define fill(x,t) memset(x,t,sizeof(x))
#define lowbit(x) (x&(-x))
using std:: swap;
const int N=200005;
const int E=400005;
struct edge{int x,y,next;}e[E];
int size[N],pos[N],bl[N],dep[N],fa[N][21];
int c[N],n;
int ls[N],edCnt=0;
int read() {
    int x=0,v=1; char ch=getchar();
    for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
    for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
    return x*v;
}
void addEdge(int x,int y) {
    e[++edCnt]=(edge){x,y,ls[x]}; ls[x]=edCnt;
    e[++edCnt]=(edge){y,x,ls[y]}; ls[y]=edCnt;
}
void modify(int x,int v) {
    for (int i=x;i<=n;i+=lowbit(i)) {
        c[i]+=v;
    }
}
int query(int x) {
    int ret=0;
    for (int i=x;i;i-=lowbit(i)) {
        ret+=c[i];
    }
    return ret;
}
void dfs1(int now) {
    rep(i,1,20) fa[now][i]=fa[fa[now][i-1]][i-1];
    size[now]=1;
    for (int i=ls[now];i;i=e[i].next) {
        if (dep[e[i].y]) continue;
        fa[e[i].y][0]=now;
        dep[e[i].y]=dep[now]+1;
        dfs1(e[i].y);
        size[now]+=size[e[i].y];
    }
}
void dfs2(int now,int up) {
    pos[now]=++pos[0];
    bl[now]=up;
    int mx=0;
    for (int i=ls[now];i;i=e[i].next) {
        if (size[e[i].y]>size[mx]&&dep[e[i].y]>dep[now]) {
            mx=e[i].y;
        }
    }
    if (!mx) return ;
    dfs2(mx,up);
    for (int i=ls[now];i;i=e[i].next) {
        if (e[i].y==mx||dep[e[i].y]<dep[now]) continue;
        dfs2(e[i].y,e[i].y);
    }
}
int get(int x,int y) {
    int ret=0;
    while (bl[x]!=bl[y]) {
        if (dep[bl[x]]<dep[bl[y]]) swap(x,y);
        ret+=query(pos[x])-query(pos[bl[x]]-1);
        x=fa[bl[x]][0];
    }
    if (pos[x]>pos[y]) swap(x,y);
    ret+=query(pos[y])-query(pos[x]-1);
    return ret;
}
int get_node(int x,int y) {
    drp(i,20,0) {
        if (y>=(1<<i)) {
            y-=(1<<i);
            x=fa[x][i];
        }
    }
    return x;
}
int main(void) {
    n=read();
    int T=read();
    rep(i,2,n) {
        int x=read(),y=read();
        addEdge(x,y);
    }
    dep[1]=1;
    dfs1(1);
    dfs2(1,1);
    modify(pos[1],1);
    while (T--) {
        char opt[2]; scanf("%s",opt);
        int x=read();
        if (opt[0]=='C') {
            if (query(pos[x])-query(pos[x]-1)==0) {
                modify(pos[x],1);
            }
        } else {
            int prt=0;
            int l=0,r=n;
            while (l<=r) {
                int mid=(l+r)>>1;
                int node=get_node(x,mid);
                int ret=get(x,node);
                if (ret>=1) prt=node,r=mid-1;
                else l=mid+1;
            }
            printf("%d\n", prt);
        }
    }
    return 0;
}