HDU 3974 Assign the task 【线段树+dfs时间戳】

题目来源：http://acm.hdu.edu.cn/showproblem.php?pid=3974

---

★开学之后又有好久没写博客了，快一个月了，写了一下以前写过的题目，居然忘记怎么写了。。。
不会写的原因是MLE 没有把head数组清0，调了一晚上，枯了

---

题意：

现在有一个公司，里面有n个员工，员工之间有上级和下属的关系。上级得到了任务 他会立刻和其所有下属转手干这份工作（把之前的事抛在一边）。然后给你两个操作，①：给某个员工一个工作 ②：问某个员工在干什么工作

---

思路：

在dfs遍历树的过程中 记录下每个父节点的起始时间和终止时间

（点左边的数字是起始时间，右边的数字是终止时间）
然后很容易发现，起始时间 和 终止时间 之间的点就是 父节点的子节点
所以我们更新的时候在这个区间更新就好了
单点查询的时候，查询 起始点 终止点 都行
剩下的就是基本的线段树操作了~详见代码

---

代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<string>
#include<map>
#include<stack>
#include<queue>
#include<set>
#include<vector>
#define ls k<<1,l,mid
#define rs k<<1|1,mid+1,r
using namespace std;
typedef long long LL;
const int maxn=5e4+5;
const int mod=1e9+7;
const int inf=0x7ffff;
const double pi=acos(-1);
const double eps=1e-8;
int n,m,cnt;
struct node
{
    int to,nex;
}g[maxn];
int head[maxn],st[maxn],en[maxn];
int v[maxn<<2],lazy[maxn<<2];
bool vis[maxn];
template<class t>
inline void read(t &x)
{
    char c;x=1;
    while((c=getchar())<'0'||c>'9') if(c=='-') x=-1;
    t res=c-'0';
    while((c=getchar())>='0'&&c<='9') res=res*10+c-'0';
    x*=res;
}
void addedge(int x,int y)
{
    g[++cnt].to=y;
    g[cnt].nex=head[x];
    head[x]=cnt;
}
void dfs(int k)
{
//    cout<<k<<'x'<<endl;
    st[k]=++cnt;
    for(int i=head[k];i;i=g[i].nex){
        dfs(g[i].to);
    }
    en[k]=cnt;
}
void build(int k,int l,int r)
{
    lazy[k]=0;
    v[k]=-1;
    if(l==r) return ;
    int mid=l+r>>1;
    build(ls); build(rs);
}
void push_down(int k,int l,int r,int mid)
{
    if(lazy[k]){
        lazy[k<<1]=lazy[k<<1|1]=lazy[k];
        v[k<<1]=v[k<<1|1]=lazy[k];
        lazy[k]=0;
    }
}
void update(int k,int l,int r,int x,int y,int val)
{
    if(x<=l&&r<=y){
        v[k]=val;
        lazy[k]=val;
        return ;
    }
    int mid=l+r>>1;
    push_down(k,l,r,mid);
    if(mid>=x) update(ls,x,y,val);
    if(mid<y) update(rs,x,y,val);
}
int query(int k,int l,int r,int pos)
{
    if(l==r) return v[k];
    int mid=l+r>>1;
    push_down(k,l,r,mid);
    if(mid>=pos) return query(ls,pos);
    return query(rs,pos);
}
int main()
{
    int t;
    read(t);
    int cas=0;
    while(t--){
        read(n);
        printf("Case #%d:\n",++cas);
        memset(vis,0,sizeof vis);
        memset(head,0,sizeof head);
        cnt=0;
        for(int i=1;i<n;i++){
            int a,b;
            read(a); read(b);
            addedge(b,a);
            vis[a]=1;
        }
        cnt=0;
        for(int i=1;i<=n;i++){
            if(!vis[i]){
                dfs(i);
                break;
            }
        }
        build(1,1,cnt);
        char s[10];
        int p,q;
        read(m);
//        for(int i=1;i<=n;i++){
//            cout<<st[i]<<' '<<en[i]<<endl;
//        }
        while(m--){
            scanf("%s %d",s,&p);
            if(s[0]=='C'){
                printf("%d\n",query(1,1,cnt,st[p]));
            }
            else{
                read(q);
                update(1,1,cnt,st[p],en[p],q);
            }
        }
    }
    return 0;
}