HDU-3974-Assign the task(线段树维护dfs序)

题目链接: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


转载于:https://www.cnblogs.com/GrowingJlx/p/6642640.html

你可能感兴趣的:(HDU-3974-Assign the task(线段树维护dfs序))