题目:一颗有根树n个节点,m次操作,T操作是以节点x为根的所有子树涂上颜色y,C操作是询问节点x是什么颜色。(1<=n,m<=1e5)
感想+题解:这道题…真的是大杂脍,为了考知识点出题人费劲苦心。就是把整棵树跑dfs序,然后操作节点x就是给dfs序中的[L[x],R[x]]每个数涂上颜色y,这个就是要用线段树的延迟标记了。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<set>
#include<vector>
#define ll long long
#define ull unsigned long long
#define ld long double
#define rep1(i,n) for(int i=1;i<=n;i++)
#define rep0(i,n) for(int i=0;i<n;i++)
#define rep(i,l,r) for(int i=l;i<=r;i++)
using namespace std;
const ll mod=999911659;
const int maxn=1e6+7,INF=0x3f3f3f3f;
#define lson p<<1
#define rson p<<1|1
int n,m,tot;
int deg[maxn],x[maxn],y[maxn];
vector<int>e[maxn];
struct S
{
int l,r;
int col;
}t[maxn*4];
void build(int p,int l,int r)
{
t[p].l=l,t[p].r=r,t[p].col=-1;
if(l==r)return;
int mid=(l+r)/2;
build(lson,l,mid);
build(rson,mid+1,r);
}
void spread(int p){
if(t[p].col!=-1){
t[lson].col=t[rson].col=t[p].col;
t[p].col=-1;
}
}
void change(int p,int l,int r,int k)
{
if(t[p].l>=l&&t[p].r<=r){
t[p].col=k;
return;
}
spread(p);
int mid=(t[p].l+t[p].r)/2;
if(l<=mid)change(lson,l,r,k);
if(r>mid)change(rson,l,r,k);
}
int ask(int p,int x)
{
if(t[p].l==t[p].r)return t[p].col;
spread(p);
int mid=(t[p].l+t[p].r)/2;
if(x<=mid)return ask(lson,x);
if(x>mid)return ask(rson,x);
}
void dfs(int p){
x[p]=++tot;
for(int i=0;i<e[p].size();i++)dfs(e[p][i]);
y[p]=++tot;
}
int main()
{
// freopen("in.txt","r",stdin);
int tt;scanf("%d",&tt);
for(int c=1;c<=tt;c++){
scanf("%d",&n);
tot=0;
memset(deg,0,sizeof(deg));
for(int i=1;i<=n;i++)e[i].clear();
for(int i=1;i<=n-1;i++){
int l,r;scanf("%d%d",&l,&r);
e[r].push_back(l);
deg[l]++;
}
int root;
for(int i=1;i<=n;i++)
if(deg[i]==0){root=i;break;}
dfs(root);
scanf("%d",&m);
build(1,1,tot);
printf("Case #%d:\n",c);
for(int i=1;i<=m;i++){
char s;cin>>s;
if(s=='C'){
int k;scanf("%d",&k);
printf("%d\n",ask(1,x[k]));
}
else{
int l,k;scanf("%d%d",&l,&k);
change(1,x[l],y[l],k);
}
}
}
return 0;
}