kb-07线段树--10--dfs序建树

/*
   hdu3974
   dfs序建树，然后区间修改查询；
 */
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define MAX_N 50005
using namespace std;

int N,M;
struct Node
{
    int to,next;
}edge[MAX_N];
struct TREE
{//val记录的是区间；
    int l,r,val,lazy;
}tr[MAX_N*4];
int head[MAX_N],cnt,tot;
int Start[MAX_N],End[MAX_N];

void Addedge(int u,int v)//链式前向星存图；
{
    //tot是边的编号；
    edge[tot].to=u;//以v为起点的边；终点是u；
    edge[tot].next=head[v];//这条边的下一条边是几号边，依然是v为起点的边；
    head[v]=tot++;//更新v为起点的边的入口；
}
void dfs(int x)
{
    cnt++;//初始为0；这里是1了，从1开始编号；
    Start[x]=cnt;//起点是x的编号
    for(int i=head[x];i!=-1;i=edge[i].next)//遍历以v为起点的边；
    {
        dfs(edge[i].to);
    }
    End[x]=cnt;//遍历完子树后可以得到这个树根的编号范围就是star&end；
    //dfs先根遍历，把x为根的子树的所有的节点编号在一段区间内，连续编号，并且记录该区间的编号起始和结束，当线段树修改值的时候就直接修改该区间；
}
void build(int rt,int l,int r)
{
    tr[rt].l=l;
    tr[rt].r=r;
    tr[rt].val=-1;
    tr[rt].lazy=0;
    if(l==r)
        return;
    int mid=(l+r)/2;
    build(rt<<1,l,mid);
    build(rt<<1|1,mid+1,r);
}
void Pushdown(int rt)
{
    int ls=rt<<1,rs=rt<<1|1;
    if(tr[rt].lazy)
    {
        tr[rs].val=tr[ls].val=tr[rt].val;
        tr[rs].lazy=tr[ls].lazy=1;
        tr[rt].lazy=0;
    }
}
void Update(int rt,int l,int r,int x)
{
    if(tr[rt].l==l&&tr[rt].r==r)
    {
        tr[rt].val=x;
        tr[rt].lazy=1;
        return ;
    }
    Pushdown(rt);
    int ls=rt<<1,rs=rt<<1|1;
    if(l<=tr[ls].r)
    {
        if(r<=tr[ls].r)
            Update(ls,l,r,x);
        else
            Update(ls,l,tr[ls].r,x);
    }
    if(r>=tr[rs].l)
    {
        if(l>=tr[rs].l)
            Update(rs,l,r,x);
        else
            Update(rs,tr[rs].l,r,x);
    }
}
int Query(int rt,int a)
{
    if(tr[rt].l==a&&tr[rt].r==a)
    {
        return tr[rt].val;
    }
    Pushdown(rt);
    int mid=(tr[rt].l+tr[rt].r)/2;
    if(a<=mid)
        return Query(rt<<1,a);
    else
        return Query(rt<<1|1,a);
}
int main()
{
    int T,k=1;
    cin>>T;
    while(T--)
    {
        printf("Case #%d:
",k++);
        cnt=0;
        tot=0;
        memset(edge,0,sizeof(edge));
        memset(head,-1,sizeof(head));
        memset(Start,-1,sizeof(Start));
        memset(End,-1,sizeof(End));
        bool used[MAX_N];
        memset(used,false,sizeof(used));
        cin>>N;
        for(int i=0;i<N-1;i++)
        {
            int u,v;
            cin>>u>>v;//v是u的上级；
            used[u]=true ;//表示u是有上级的，也就是有入度的；
            Addedge(u,v);
        }
        for(int i=1;i<=N;i++)
        {
            if(!used[i])//找到入度为0的点，就是整个树的入口；
            {
                dfs(i);
                break;
            }
        }
        build(1,1,N);
        cin>>M;
        for(int i=0;i<M;i++)
        {
            char a[10];
            scanf("%s",a);
            if(a[0]=='C')
            {
                int x;
                cin>>x;
                int ans=Query(1,Start[x]);//该点的编号就是起点的编号；
                cout<<ans<<endl;
            }
            else
            {
                int a,b;
                cin>>a>>b;
                Update(1,Start[a],End[a],b);
            }
        }

    }
    return 0;
}