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<5105,y<109 N < 5 ∗ 10 5 , y < 10 9


思路

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

这里,就有一个神奇的算法,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));
            }
        }
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值