[BZOJ4551][Tjoi2016&Heoi2016]树（dfs序+线段树）

题目描述

传送门

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

题解

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

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
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]]<h[x]) maxn[now]=x;
        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]<h[maxn[now]]) ans=maxn[now];
    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;i<n;++i)
    {
        int 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);
        }
    }
}