BZOJ4551: [Tjoi2016&Heoi2016]树

BZOJ4551

题意很清晰。。也很好做。。第一次对拍一次过。。
树链剖分，线段树里维护一个区间内最右边的被标记过的点的位置就可以了。据说数据很水。。暴力一步一步爬可以拿满分！！？

【代码】

#include <iostream>
#include <cstdio>
#include <stack>
#include <cstring>
#include <algorithm>
#include <cmath>
#define N 100005
using namespace std;
typedef long long ll;
typedef pair<int,int> pa;

ll read()
{
    ll x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
} 

int n,m,cnt,ind;
bool mark[N];
int b[N<<1],p[N],nextedge[N<<1];
int fa[N],son[N],deep[N],top[N],pos[N],ftree[N],sz[N];

class Seg_Tree{
    public:
        int l,r,pos;
        bool Flag;
}e[N<<2];

void Add(int x,int y)
{
    cnt++;
    b[cnt]=y;
    nextedge[cnt]=p[x];
    p[x]=cnt;
}

void Anode(int x,int y){
    Add(x,y);Add(y,x);
}

void Dfs(int x)
{
    sz[x]=1;
    for(int i=p[x];i;i=nextedge[i])
    {
        int v=b[i];
        if(v==fa[x]) continue;
        fa[v]=x;deep[v]=deep[x]+1;
        Dfs(v);sz[x]+=sz[v];
        son[x]=sz[son[x]]>sz[v]?son[x]:v;
    }
}

void dfs(int x,int root)
{
    pos[x]=++ind;top[x]=root;
    ftree[pos[x]]=x;
    if(!son[x]) return;
    dfs(son[x],root);
    for(int i=p[x];i;i=nextedge[i])
    {
        int v=b[i];
        if(v==fa[x]||v==son[x]) continue;
        dfs(v,v);
    }
}

void pushup(int p){
    e[p].Flag=e[p<<1].Flag|e[p<<1|1].Flag;
    if(e[p<<1|1].pos) e[p].pos=e[p<<1|1].pos;
    else e[p].pos=e[p<<1].pos;
}

void Build(int p,int l,int r)
{
    e[p].l=l,e[p].r=r;if(l==1) e[p].Flag=1,e[p].pos=1;
    if(l==r) return;
    int mid=l+r>>1;
    Build(p<<1,l,mid);Build(p<<1|1,mid+1,r);
}

void Input_Init()
{
    n=read(),m=read();
    for(int i=1;i<n;i++)
    {
        static int x,y;
        x=read(),y=read();
        Anode(x,y);
    }
    Dfs(1);dfs(1,1);Build(1,1,n);
}

void Get_Tag(int p,int x)
{
    int l=e[p].l,r=e[p].r,mid=l+r>>1;
    if(l==r&&l==x){
        e[p].Flag=1;e[p].pos=l;
        return;
    }
    if(x<=mid) Get_Tag(p<<1,x);
    else Get_Tag(p<<1|1,x);
    pushup(p);
}

int Query_Pos(int p,int x,int y)
{
    int l=e[p].l,r=e[p].r,mid=l+r>>1;
    if(!e[p].Flag) return 0;
    if(l==x&&y==r)
        return e[p].pos;
    if(y<=mid) return Query_Pos(p<<1,x,y);
    if(x>mid) return Query_Pos(p<<1|1,x,y);
    int Right=Query_Pos(p<<1|1,mid+1,y);
    if(Right) return Right;
    return Query_Pos(p<<1,x,mid);
}

void Query(int x)
{
    int fx=top[x];
    while(true)
    {
        int rtn=Query_Pos(1,pos[fx],pos[x]);
        if(rtn){
            printf("%d\n",ftree[rtn]);
            return;
        }
        x=fa[fx];fx=top[x];
    }
}

void Solve()
{
    mark[1]=1;
    for(int i=1;i<=m;i++)
    {
        char ch[2];scanf("%s",ch);
        static int x;x=read();
        if(ch[0]=='C') {
            if(!mark[x]) Get_Tag(1,pos[x]),mark[x]=1;
        }
        else Query(x);
    }
}

int main()
{
    Input_Init();
    Solve();
    return 0;
}