hdu 3974

这题的关键在于dfs建序

 

这样2 对应的区间就是1-5

3对应的区间就是2-4

4对应的区间就是3-3

1对应的区间就是4-4

5对应的区间就是5-5

剩下的就是区间修改 单点查询了

代码如下

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=50005;
int s[maxn],e[maxn];
int index;
vector<int> g[maxn];
bool used[maxn];
int x,y;
int sum[maxn*4],lazy[maxn*4];
void dfs(int k)
{
    s[k]=++index;
    for(int i=0,len=g[k].size();i<len;i++) {
        dfs(g[k][i]);
    }
    e[k]=index;
}
void Build(int o,int l,int r)
{
    sum[o]=-1; lazy[o]=-1;
    if(l==r) {
        return ;
    }
    int mid=l+r>>1;
    Build(o<<1,l,mid);
    Build(o<<1|1,mid+1,r);
}
void Push_down(int o,int l,int r)
{
    if(lazy[o]>=0) {
        sum[o<<1]=sum[o<<1|1]=lazy[o<<1]=lazy[o<<1|1]=lazy[o];
        lazy[o]=-1;
    }
}
void Update(int o,int l,int r)
{
    if(s[x]<=l&&r<=e[x]) {
        sum[o]=y; lazy[o]=y; return ;
    }
    Push_down(o,l,r);
    int mid=l+r>>1;
    if(s[x]<=mid) {
        Update(o<<1,l,mid);
    }
    if(e[x]>mid) {
        Update(o<<1|1,mid+1,r);
    }
}
int Query(int o,int l,int r)
{
    if(l==r) {
        return sum[o];
    }
    Push_down(o,l,r);
    int mid=l+r>>1;
    if(s[x]<=mid) {
        return Query(o<<1,l,mid);
    }
    else{
        return Query(o<<1|1,mid+1,r);
    }
}
int main()
{
    int t,n,m;
    scanf("%d",&t);
    char s[5];
    int cnt=0;
    while(t--) {
        scanf("%d",&n);
        for(int i=1;i<=n;i++) {
            g[i].clear();
        }
        memset(used,false,sizeof(used));
        for(int i=1;i<n;i++) {
            int u,v;
            scanf("%d%d",&u,&v);
            g[v].push_back(u);
            used[u]=true;
        }
        index=0;
        for(int i=1;i<=n;i++) {
            if(!used[i]) {
                dfs(i); break;
            }
        }
        Build(1,1,n);
        printf("Case #%d:\n",++cnt);
        scanf("%d",&m);
        while(m--) {
            scanf("%s",s);
            if(s[0]=='T') {
                scanf("%d%d",&x,&y);
                Update(1,1,n);
            }
            else{
                scanf("%d",&x);
                printf("%d\n",Query(1,1,n));
            }
        }
    }
}