题目链接:HDU-3974-Assign the task
题意是给定一棵树,然后一种操作是指定一个点,这个点及这个点的的子树被染色,另一种操作是指定一个点,问这个点的颜色。
可以通过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