J - Assign the task（线段树+dfs序）

题目链接

通过dfs序 将 上下级关系树 转换成线性结构， （转换成线性结构 便于 上级 和下级一起更新）

然后构建线段树， 进行修改查询。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 5e5+50;

int dfs[maxn]; // dfs序数组
vector<int>G[maxn];//图
int st[maxn];//i  子树在dfs序中开始的位置
int ed[maxn];// i 子树 结束的位置
int len;//dfs数组的长度
int tim;// 时间
int pos[maxn]; // i在dfs序中的位置

bool d[maxn];// 标记是否有入度


struct node
{
    int l;
    int r;
    int lazy;
    int work;

}Segtr[maxn*4];

void DFS(int x, int l)
{
    dfs[++len] = x;
    st[x] = ++tim;
    pos[x] = len;

    for(int i = 0;i < (int)G[x].size();i++)
    {

            DFS(G[x][i],x);

    }

    ed[x] = tim;

}
void build(int rt, int l, int r)
{
    Segtr[rt].l = l;
    Segtr[rt].r = r;
    Segtr[rt].lazy  = -1;
    Segtr[rt].work = -1;

    if(l == r)
    {
        return;
    }
    int mid = (l+r)>>1;
    build(rt*2+1, l, mid);
    build(rt*2+2, mid+1, r);
}
void init(int n)
{
    len = 0;
    tim = 0;
    for(int i = 0;i <= n;i++)G[i].clear();//
    memset(d, false , sizeof(d));//
}
void pushdown(int rt)
{
    Segtr[rt*2+1].work = Segtr[rt].lazy;
    Segtr[rt*2+2].work = Segtr[rt].lazy;
    Segtr[rt*2+1].lazy = Segtr[rt*2+2].lazy = Segtr[rt].lazy;
    Segtr[rt].lazy = -1;

}
void update(int rt, int L, int R, int y)
{
    int l = Segtr[rt].l;
    int r = Segtr[rt].r;

    if(l == L && R == r)
    {
        Segtr[rt].lazy = y;
        Segtr[rt].work = y;
        return;
    }
    if(Segtr[rt].lazy != -1)
    {
        pushdown(rt);
    }

    int mid = (l+r)>>1;

    if(R<= mid)
    {
        update(rt*2+1, L, R , y);
    }
    else if (L > mid)
    {
        update(rt*2+2, L, R, y);
    }
    else
    {
        update(rt*2+1, L, mid, y);
        update(rt*2+2, mid+1, R, y);
    }


}
int query(int rt ,int x)
{
    int l = Segtr[rt].l;
    int r=  Segtr[rt].r;

    if(l == r)
    {
        return Segtr[rt].lazy;
    }
    int ret;

    if(Segtr[rt].lazy!= -1)
    {
        pushdown(rt);
    }

    int mid = (l + r )>>1;

    if(x <= mid)
    {
        ret = query(rt*2+1, x);
    }
    else if(x > mid)
    {
        ret = query(rt*2+2, x);
    }

    return ret;
}
int main()
{

    int T, cnt = 0;//
    cin>>T;//

    while(T--)
    {
        int n;//
        cin>>n;//
        build(0 , 1 , n);
        init(n);

        for(int i = 0;i < n-1;i++)//
        {
            int u, v;
            cin>>u>>v;
            G[v].push_back(u);
            d[u] = 1;
        }

        for(int i = 1;i <= n;i++)
        {
            if(!d[i])
            {
                DFS(i, -1);
                break;
            }
        }

        int m;
        cin>>m;
        char op;
        int x, y;
        cout <<"Case #"<< ++cnt<<":"<<endl;
        while(m--)
        {
            cin>>op;

            if(op == 'C')
            {
                cin>>x;
                cout <<query(0, pos[x])<<endl;// 按照dfs序中的位置查询
            }
            else
            {
                cin>>x>>y;
                update(0, st[x], ed[x], y);// 更新左右子树
            }

        }

    }


    return 0;
}