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

来源:互联网 发布:淘宝电动清扫车多少钱 编辑:程序博客网 时间:2024/05/20 18:52

题目描述

传送门

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

题解

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

代码

#include<algorithm>#include<iostream>#include<cstring>#include<cstdio>#include<cmath>using namespace std;#define N 100005int 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);        }    }}
0 0
原创粉丝点击