HDU-3974 Assign the task （DFS序+线段树）

Assign the task

题目链接： J - Assign the task HDU - 3974

题意

有一个公司，一共有N个员工，每个员工有一个直接上司（董事长没有），每个上司可以管理自己手下以及自己手下的所有手下。关系为N-1个，有两个操作。

1. C 查询x员工正在干的事
2. T 让x员工及其手下干y事

数据范围：$N<5 *10^5,y < 10^9$

---

思路

因为数据量非常的大，所以不可能每次都从根节点一步一步扫下来的，一定要转换成一个复杂度为$O(logN)$ 的查询方式修改方式才可以。这明显可以区间修改，单点查询。可是怎么转换成线段树呢？因为其区间并不是连续的。

这里，就有一个神奇的算法，dfs序，就是正常dfs时所记录的访问次序，每个节点的访问次序，必定与其子节点的访问次序相连接，那么怎么找到区间界呢，就再开一个数组e记录区间结束的地方就可以了。

void dfs(int u){  //因为这里是有向边，所以可以不用used数组，fa也不需要。
    s[u] = ++tim; //区间开始地方
    rep(i,0,E[u].size()-1) {
        dfs(E[u][i]);
    }
    e[u] = tim;   //区间结束的地方
}

---

代码

#include <bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = (int)j;i <= (int)k;i ++)
#define per(i,j,k) for(int i = (int)j;i >= (int)k;i --)
#define debug(x) cerr<<#x<<" = "<<(x)<<endl
#define mmm(a,b) memset(a,b,sizeof(a))
#define pb push_back

typedef double db;
typedef long long ll;
const int MAXN = (int)5e5+7;
const int INF = (int)0x3f3f3f3f;

vector<int> E[MAXN];
int N,degree[MAXN],tim,s[MAXN],e[MAXN];

void dfs(int u){  //因为这里是有向边，所以可以不用used数组，fa也不需要。
    s[u] = ++tim; //区间开始地方
    rep(i,0,E[u].size()-1) {
        dfs(E[u][i]);
    }
    e[u] = tim;   //区间结束的地方
}

#define lson rt<<1
#define rson rt<<1|1

int sum[MAXN<<2];
int add[MAXN<<2];

void Build(int l,int r,int rt ) {
    if (l == r) {
        sum[rt] = -1;
        add[rt] = -1;
        return ;
    }
    int m = l+r>>1;
    Build(l,m,lson);
    Build(m+1,r,rson);
    add[rt] = add[lson];
}

void Update(int L,int R,int C,int l,int r,int rt) {
    if (L <= l && r <= R) {
        sum[rt] = C;
        add[rt] = C;
        return ;
    }
    if (add[rt] != -1) sum[lson] = sum[rson] = add[lson] = add[rson] = add[rt],add[rt] = -1;
    int m = l+r>>1;
    if (L <= m) Update(L,R,C,l,m,lson);
    if (R >  m) Update(L,R,C,m+1,r,rson);
}

int Query(int L,int l,int r,int rt) {
    if (l == r) {
        return sum[rt];
    }
    if (add[rt] != -1) sum[lson] = sum[rson] = add[lson] = add[rson] = add[rt],add[rt] = -1;
    int m = l+r>>1;
    if (L <= m) return Query(L,l,m,lson);
    else        return Query(L,m+1,r,rson);
}

void debugt(int l,int r,int rt){
    if (l == r) {
        cout << l << ':' << sum[rt] << ' ';
        return ;
    }
    int m = l+r>>1;
    debugt(l,m,lson);
    debugt(m+1,r,rson);
}

void init() {
    tim = 0;
    rep(i,1,N) E[i].clear();
    rep(i,1,N) degree[i] = 0;
}

int main()
{
    int T;
    scanf("%d",&T);
    rep(ca,1,T) {
        scanf("%d",&N);
        init();
        rep(i,1,N-1){
            int u,v;
            scanf("%d %d",&u,&v);
            E[v].pb(u);
            degree[u] ++;
        }
        rep(i,1,N) if(degree[i] == 0) {dfs(i); }
        Build(1,N,1);

        printf("Case #%d:\n",ca);
        int Q;scanf("%d",&Q);
        rep(i,1,Q) {
            char op[2];int x,y;
            scanf("%s",op);
            if (op[0] == 'T'){
                scanf("%d %d",&x,&y);
                Update(s[x],e[x],y,1,N,1);
            }else {
                scanf("%d",&x);
                printf("%d\n",Query(s[x],1,N,1));
            }
        }
    }
}