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

HDU - 3974

题意：有个公司有一些关系，每个人（除了boss）都有且仅有一个上司，这就是一棵树的关系，然后会有一些操作，C i，询问第i个人现在的任务，T x y，把y任务给x， 给x相当于给x及他所有的下属，新任务会覆盖旧任务

思路：这是我写的第一个（除了联通图那些）需要用dfs序的题，最近有毒，wa到不敢相信自己....

因为i节点的子树在dfs访问序列中肯定是连续的，所以可以把dfs序处理出来，然后就可以上线段树了。

代码：

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;
const int maxn = 5e4 + 10;
struct node
{int to,next;}edge[maxn];

int h[maxn],deg[maxn];
int dfn[maxn],L[maxn], R[maxn];
int tag[maxn << 2];
int n,num,tot;

void add(int u,int v)
{
    edge[num].to = v; edge[num].next = h[u];
    h[u] = num++;
}
void init()
{
    memset(dfn,0,sizeof(dfn));
    memset(tag,0,sizeof(tag));
    for(int i = 1; i <= n; i++)
        deg[i] = 0, h[i] = -1;
    num = 0;    tot = 0;
}
void dfs(int u)     //dfs序处理一下
{
    dfn[u] = ++tot; L[u] = tot;
    for(int i = h[u]; ~i; i = edge[i].next) dfs(edge[i].to);
    R[u] = tot;
}
void pushdown(int id,int tagv)
{
    tag[id<<1] = tag[id<<1|1] = tagv;
    tag[id] = 0;
}
void updata(int id,int l,int r,int ul,int ur,int val)
{
    if(l > ur || r < ul) return;
    if(l >= ul && r <= ur){tag[id] = val;return;}
    if(tag[id]) pushdown(id,tag[id]);
    int mid = l+r>>1;
    if(ul <= mid) updata(id << 1 , l , mid, ul , ur, val);
    if(ur >  mid) updata(id << 1 | 1, mid + 1, r, ul, ur, val);
}
int query(int id,int l,int r,int u)
{
    if(l == r && l == u) return tag[id];
    if(tag[id]) pushdown(id,tag[id]);
    int mid = l+r>>1;
    if(u <= mid) return query(id << 1, l , mid, u);
    else return query(id << 1 | 1 , mid + 1 , r , u);
}
int main()
{
    int t,m,kase = 1;char str[5]; int a,b;
    scanf("%d",&t);
    while(t--)
    {
        int root = -1;
        scanf("%d",&n);
        init();
        for(int i = 0; i < n-1; i++)
        {
            scanf("%d%d",&a,&b);
            add(b,a);   deg[a]++;
        }
        for(int i = 1; i <= n; i++)
            if(deg[i] == 0) { root = i; break; }
        dfs(root);
        scanf("%d",&m);
        printf("Case #%d:\n",kase++);
        for(int i = 0; i < m; i++)
        {
            scanf(" %s%d",str,&a);
            if(str[0] == 'T')
            {
                scanf("%d",&b);
                updata(1,1,n,L[a],R[a],b + 1);
                continue;
            }
            int ans = query(1,1,n,dfn[a]);
            ans == 0? printf("-1\n") : printf("%d\n",ans-1);
        }
    }
	return 0;
}