[BZOJ4551][Tjoi2016&Heoi2016]树(dfs序+线段树)

题目描述

传送门

题目大意:给定一颗有根树(根为1),有以下两种操作:1. 标记操作:对某个结点打上标记(在最开始,只有结点1有标记,其他结点均无标记,而且对于某个结点,可以打多次标记。)2. 询问操作:询问某个结点最近的一个打了标记的祖先(这个结点本身也算自己的祖先)。

题解

16省选竟然有无脑题。。
搞出dfs序了之后就是线段树区间修改单点查询最大值问题
学习了一下标记永久化

代码

#include
#include
#include
#include
#include
using namespace std;
#define N 100005

int n,q,dfs_clock,ans;
int tot,point[N],nxt[N<<1],v[N<<1];
int h[N],in[N],out[N],maxn[N<<2];

void add(int x,int y){++tot;nxt[tot]=point[x];point[x]=tot;v[tot]=y;}
void dfs(int x,int fa)
{
    in[x]=++dfs_clock;h[x]=h[fa]+1;
    for (int i=point[x];i;i=nxt[i])
        if (v[i]!=fa)
            dfs(v[i],x);
    out[x]=dfs_clock;
}
void change(int now,int l,int r,int lr,int rr,int x)
{
    int mid=(l+r)>>1;
    if (lr<=l&&r<=rr)
    {
        if (h[maxn[now]]return;
    }
    if (lr<=mid) change(now<<1,l,mid,lr,rr,x);
    if (mid+1<=rr) change(now<<1|1,mid+1,r,lr,rr,x);
}
void query(int now,int l,int r,int x)
{
    int mid=(l+r)>>1;
    if (h[ans]if (l==r) return;
    if (x<=mid) query(now<<1,l,mid,x);
    else query(now<<1|1,mid+1,r,x);
}
int main()
{
    scanf("%d%d",&n,&q);
    for (int i=1;iint x,y;scanf("%d%d",&x,&y);
        add(x,y),add(y,x);
    }
    dfs(1,0);maxn[1]=1;
    while (q--)
    {
        char opt=getchar();
        while (opt!='C'&&opt!='Q') opt=getchar();
        int x;scanf("%d",&x);
        if (opt=='C') change(1,1,n,in[x],out[x],x);
        else
        {
            ans=0;
            query(1,1,n,in[x]);
            printf("%d\n",ans);
        }
    }
}

你可能感兴趣的:(题解,线段树,省选,dfs序)