HDU - 3974 Assign the task DFS序建线段树

训练摸鱼发现好久没写博客了，下个月去区域赛了，有点小慌，水个博客压压惊。

不想贴题目，简单说下题意就是有一群员工具有上下级关系，体现出来就是一颗多叉树，子节点是父亲的下属这样子。然后现在安排工作，更新一个员工的工作序号的时候需要同时把下属的工作序号一起完成更新。

dfs序+线段树区间处理一下就水过了

线段树是基本操作。dfs序就是因为是一颗多叉树，那么对于树dfs遍历节点的出入得到一个序列，这个序列满足一个特性，某节点的出入时间在区间的位置夹住了它所有子节点的出入时间。这样就是一道很简单的线段树区间修改单点查询的水题了。

#include<iostream>
#include<algorithm>
#include<math.h>
#include<vector>

using namespace std;

#define ll long long
const int MAXN = 5e4+5;

struct node
{
    int left,right;
    int value;
    int id;
}tree[MAXN<<3];

int fa[MAXN<<2];
int add[MAXN<<2];
int be[MAXN<<2],en[MAXN<<2];

void pushup(int i)
{
    tree[i].value = tree[i<<1].value + tree[(i<<1)+1].value;
}

void pushdown(int i)
{
    if(add[i])
    {
        add[i<<1] = add[i];
        add[(i<<1)+1] = add[i];
        tree[i<<1].value = add[i];
        tree[(i<<1)+1].value = add[i];
        add[i] = 0;
    }
}

void buildtree(int i,int l,int r)
{
    add[i] = 0;
    tree[i].left = l;
    tree[i].right = r;
    tree[i].value = -1;
    if(tree[i].left == tree[i].right)
    {
        fa[l]=i;
        return;
    }
    buildtree(i<<1,l,(int)floor((l+r)/2.0));
    buildtree((i<<1)+1,(int)floor((l+r)/2.0)+1,r);
}

void update(int i,int l,int r,int c)
{
    //cout << i << endl;
    if(tree[i].left == l && tree[i].right == r)
    {
        add[i] = c;
        tree[i].value = add[i];
        return;
    }
    pushdown(i);
    int m = (int)floor(tree[i].left + tree[i].right)/2.0;
    if(l >= m + 1)
    {
        update((i<<1)+1,l,r,c);
    }
    else if(r <= m)
    {
        update((i<<1),l,r,c);
    }
    else
    {
        update(i<<1,l,m,c);
        update((i<<1)+1,m + 1,r,c);
    }
}

ll sum = 0;

void query(int i,int l)
{
    //cout << i << endl;
    if(tree[i].left == l && tree[i].right == l)
    {
        sum = tree[i].value;
        return;
    }
    pushdown(i);
    int m = (int)floor(tree[i].left + tree[i].right)/2.0;
    if(l >= m + 1)
    {
        query((i<<1)+1,l);
    }
    else
    {
        query((i<<1),l);
    }
}

vector <int> boss[MAXN];
int tot = 0;

void dfs(int x)
{
    tot++;
    be[x] = tot;
    tree[tot].id = x;
    for(int i = 0; i < boss[x].size(); i++)
    {
        dfs(boss[x][i]);
    }
    en[x] = tot;
}

int main()
{
    ios::sync_with_stdio(false);
    int t;
    int u = 0;
    cin >> t;
    while(t--)
    {
        tot = 0;
        u++;
        cout << "Case #" << u << ":" << endl;
        int n;
        cin >> n;
        for(int i = 1; i<=n; i++)
            boss[i].clear();
        int vis[n+5];
        for(int i = 1; i <= n; i++)
            vis[i] = 0;
        for(int i = 1; i <= n - 1; i++)
        {
            int a1,a2;
            cin >> a1 >> a2;
            vis[a1] = 1;
            boss[a2].push_back(a1);
        }
        for(int i = 1; i <= n; i++)
        {
            if(vis[i] == 0)
            {
                dfs(i);
                break;
            }
        }
        buildtree(1,1,tot);
        int q;
        cin >> q;
        for(int i = 1; i <= q; i++)
        {
            char op;
            cin >> op;
            if(op == 'C')
            {
                int x;
                cin >> x;
                sum = 0;
                query(1, be[x]);
                cout << sum << endl;
            }
            else
            {
                int x1,x2;
                cin >> x1 >> x2;
                update(1,be[x1],en[x1],x2);
            }
        }
    }
    return 0;
}

本巨蒻写了半天好丑啊，本来打算这两天搞树链剖分和划分树的，但是发现还有好多知识没写啊（逃