题目链接:
题意是给定一棵树,然后一种操作是指定一个点,这个点及这个点的的子树被染色,另一种操作是指定一个点,问这个点的颜色。
可以通过dfs树将这棵树放在线段上,记结点u在线段上的位置是pos[u],以结点u为根的子树的结点总数是cnt[u],那么在每次染色就是染色线段上pos[u]~pos[u]+cnt[u]-1这段区间。
然后用线段树维护就行了。
#include#include #include #include using namespace std;const int maxn=5e4+7;int n,m;vector adj[maxn];int cnt[maxn],pos[maxn],c[maxn<<2];bool vis[maxn];int dfs(int u){ pos[u]=++m; int res=1; for(int i=0;i =r) { c[now]=t; return ; } push_down(now); int mid=(l+r)>>1; if(ul<=mid) update(now<<1,l,mid,ul,ur,t); if(ur>mid) update(now<<1|1,mid+1,r,ul,ur,t);}int query(int now,int l,int r,int p){ if(l==r) return c[now]; int mid=(l+r)>>1; push_down(now); if(p<=mid) return query(now<<1,l,mid,p); else return query(now<<1|1,mid+1,r,p);}int main(){ int T,kase=1; scanf("%d",&T); while(T--) { int u,v; scanf("%d",&n); for(int i=1;i<=n;i++) adj[i].clear(); memset(cnt,0,sizeof(cnt)); for(int i=1;i