bzoj 4551: [Tjoi2016&Heoi2016]树 （线段树）

题目描述

传送门

题解

感觉这道题写法其实应该挺多的，我写的线段树的区间修改单点查询，然后用的标记永久化。。

代码

#include<iostream> #include<cstdio> #include<algorithm> #include<cstring> #define N 200003 using namespace std; int tot,point[N],nxt[N],v[N],l[N],r[N]; int tr[N*4],delta[N*4],deep[N],n,m,sz; int add(int x,int y) { tot++; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; tot++; nxt[tot]=point[y]; point[y]=tot; v[tot]=x; } void dfs(int x,int fa) { deep[x]=deep[fa]+1; l[x]=++sz; for (int i=point[x];i;i=nxt[i]) { if (v[i]==fa) continue; dfs(v[i],x); } r[x]=sz; } void build(int now,int l,int r) { if (l==r) { tr[now]=1; delta[now]=1; return; } int mid=(l+r)/2; build(now<<1,l,mid); build(now<<1|1,mid+1,r); } void query(int now,int l,int r,int ll,int rr,int x) { if (ll<=l&&r<=rr) { if(deep[delta[now]]<deep[x]) delta[now]=x; return; } int mid=(l+r)/2; if (ll<=mid) query(now<<1,l,mid,ll,rr,x); if (rr>mid) query(now<<1|1,mid+1,r,ll,rr,x); } int find(int now,int l,int r,int x) { if (l==r) return delta[now]; int mid=(l+r)/2; int ans=delta[now]; int t=0; if (x<=mid) t=find(now<<1,l,mid,x); else t=find(now<<1|1,mid+1,r,x); if (deep[ans]>deep[t]) return ans; else return t; } int main() { freopen("a.in","r",stdin); scanf("%d%d",&n,&m); for (int i=1;i<n;i++) { int x,y; scanf("%d%d",&x,&y); add(x,y); } dfs(1,0); build(1,1,n); for (int i=1;i<=m;i++) { char s[5]; int x; scanf("%s%d",s+1,&x); if (s[1]=='C') query(1,1,n,l[x],r[x],x); else printf("%d\n",find(1,1,n,l[x])); } }