J - Assign the task

J - Assign the task

HDU - 3974

思路：一眼秒思路<(*￣▽￣*)/

dfs序+线段树。

通过dfs序把树上问题转化成线段上的问题。然后用线段树解决。

 

 错因：都是些zz的错误就不说了
 

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAXN 50010
using namespace std;
int t,n,m;
int sz,tot,sum,root;
int to[MAXN*],net[MAXN*],head[MAXN];
int dad[MAXN],idl[MAXN],idr[MAXN],edge[MAXN];
struct nond{
    int l,r,flag;
}tree[MAXN*];
void add(int u,int v){
    to[++tot]=v;net[tot]=head[u];head[u]=tot;
}
void build(int now,int l,int r){
    tree[now].l=l;tree[now].r=r;
    tree[now].flag=-;
    if(tree[now].l==tree[now].r)    return ;
    int mid=(tree[now].l+tree[now].r)/;
    build(now*,l,mid);
    build(now*+,mid+,r);
}
void down(int now){
    tree[now*].flag=tree[now*+].flag=tree[now].flag;
    tree[now].flag=-;
    return ;
}
void change(int now,int l,int r,int pos){
    if(tree[now].l==l&&tree[now].r==r){
        tree[now].flag=pos;
        return ;
    }
    if(tree[now].flag!=-)    down(now);
    int mid=(tree[now].l+tree[now].r)/;
    if(r<=mid)    change(now*,l,r,pos);
    else if(l>mid)    change(now*+,l,r,pos);
    else { change(now*,l,mid,pos);change(now*+,mid+,r,pos); }
}
int query(int now,int x){
    if(tree[now].l==tree[now].r)
        return tree[now].flag;
    if(tree[now].flag!=-)    down(now);
    int mid=(tree[now].l+tree[now].r)/;
    if(x<=mid)    return query(now*,x);
    else return query(now*+,x);
}
void dfs(int now){
    idl[now]=++sz;
    for(int i=head[now];i;i=net[i])
        if(dad[now]!=to[i]){
            dad[to[i]]=now;
            dfs(to[i]);
        }
    idr[now]=++sz;
}
int main(){
    scanf("%d",&t);
    while(t--){
        sum++;printf("Case #%d:\n",sum);
        scanf("%d",&n);tot=;
        for(int i=;i<n;i++){
            int u,v;
            scanf("%d%d",&u,&v);
            add(v,u);edge[u]++;
        }
        for(int i=;i<=n;i++)
            if(edge[i]==){ root=i;break; }
        dfs(root);build(,,sz);
        scanf("%d",&m);
        for(int i=;i<=m;i++){
            char c;int x,y;
            scanf("\n%c%d",&c,&x);
            if(c=='T'){
                scanf("%d",&y);
                change(,idl[x],idr[x],y);
            }
            else    printf("%d\n",query(,idl[x]));
        }
        root=;sz=;
        memset(to,,sizeof(to));
        memset(idl,,sizeof(idl));
        memset(idr,,sizeof(idr));
        memset(net,,sizeof(net));
        memset(dad,,sizeof(dad));
        memset(edge,,sizeof(head));
        memset(head,,sizeof(head));
    }
}
/*
1
5
4 3
3 2
1 3
5 2
5
C 3
T 2 1
 C 3
T 3 2
C 3
*/