HDU-3974-Assign the task（线段树维护dfs序）-优快云博客

本文介绍了一道关于树结构的数据结构题目HDU-3974-Assignthetask的解决方法。通过深度优先搜索（DFS）将树转化为线段树，再利用线段树进行区间更新和查询操作，实现了树节点的染色和颜色查询。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意是给定一棵树，然后一种操作是指定一个点，这个点及这个点的的子树被染色，另一种操作是指定一个点，问这个点的颜色。

可以通过dfs树将这棵树放在线段上，记结点u在线段上的位置是pos[u]，以结点u为根的子树的结点总数是cnt[u]，那么在每次染色就是染色线段上pos[u]~pos[u]+cnt[u]-1这段区间。

然后用线段树维护就行了。

#include<cstdio>
#include<algorithm>
#include<vector>
#include<cstring>
using namespace std;
const int maxn=5e4+7;
int n,m;
vector<int> 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<adj[u].size();i++)
    {
        int v=adj[u][i];
        if(vis[v])  continue;
        vis[v]=true;
        res+=dfs(v);
    }
    return cnt[u]=res;
}

void push_down(int now)
{
    if(c[now]==-1)  return ;
    c[now<<1]=c[now<<1|1]=c[now];
    c[now]=-1;
}

void update(int now,int l,int r,int ul,int ur,int t)
{
    if(ul<=l&&ur>=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<n;i++)
        {
            scanf("%d%d",&u,&v);
            adj[v].push_back(u);
            cnt[u]++;
        }
        for(int i=1;i<=n;i++)
        {
            if(!cnt[i])
            {
                m=0;
                memset(cnt,0,sizeof(cnt));
                memset(vis,false,sizeof(vis));
                dfs(i);
                break;
            }
        }
        memset(c,-1,sizeof(c));
        scanf("%d",&m);
        char option[3];
        printf("Case #%d:\n",kase++);
        for(int i=0;i<m;i++)
        {
            scanf("%s",option);
            if(option[0]=='C')
            {
                scanf("%d",&u);
                printf("%d\n",query(1,1,n,pos[u]));
            }
            else
            {
                scanf("%d%d",&u,&v);
                update(1,1,n,pos[u],pos[u]+cnt[u]-1,v);
            }
        }
    }
}

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