HDU - 3974 Assign the task（线段树+dfs序）

题意：给你一个树形结构，两种操作，第一种是把一个结点染色，这个结点的所有子结点都要染成同样的颜色。第二种操作是查询某个结点的颜色。这让我们想到线段树，但是线段树是一维直线的，怎么解决树形结构呢？

先介绍一下dfs序，大家都知道dfs搜索顺序，从根结开始不断向叶子结点单向搜索，如果到叶子结点就向上回溯一层，继续向其他子结点搜索。这就造成了一种特性，就是搜索完所有子结点会再回到父结点，再搜完父结点的兄弟结点再回到祖先结点。这是一种非常舒服的搜索顺序。跟这道题什么关系呢？在看到这一题说道要给一个结点染色，就会给这个子树染色，那我们为什么不直接把这个子树当成区间呢。我们只需要用dfs序输出这个树，每个父结点都会连着一段为其子树的区间。直接用线段树进行维护就行了。

#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);
            }
        }
    }
}