Assign the task(DFS序)

题目链接:Assign the task

大致题意

推荐看CaprYang的版本, 中文题面很棒, 唯一的问题就是, 输入图的时候是n-1行, 而不是n行.

解题思路

既然是线段树专题, 我们的解题方法就仅提供线段树做法.

这个题不同于其他题的点就在于, 我们需要维护的不是一个给出的区间, 而是一棵树. 而我们的线段树维护的理论上应该是一个区间. 所以本题的难点在于, 如何把一棵树区间化.

这里推荐大家先去学习一下 dfs序, 我们可以通过dfs序来给每个节点分配一个新的编号, 从而再将整棵树区间化.

由于这个题修改和查询操作都比较简单, 考察的是子树修改和单点查询. 因为通过dfs序列得到的子树内部节点编号一定是连续的, 所以我们可以通过记录dfs序的时间戳来解决本题. 即: 记录每棵树的进入时间和离开时间(时间戳的值需保证与节点编号值同步) (相当于新的节点编号==进入当前节点的时间)

注: 上图为本题样例的解释说明

---

此外如果本题复杂一些, 例如要修改2 ~ 3这一条路径上的值, 我们就需要通过树链剖分来解决了.

AC代码(DFS序时间戳)

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
using namespace std;
typedef long long ll;
const int N = 5E4 + 10;
int n, m;
vector<int> edge[N];
int BG[N], ED[N], ind; //进入时间戳(节点的新编号), 结束时间戳
int du[N];

void dfs(int x) {
    BG[x] = ++ind;
    for (auto& to : edge[x]) dfs(to); 
    ED[x] = ind;
}


struct node {
    int l, r;
    int col;
    bool lazy;
}t[N << 2];

void pushdown(node& op, int col) { op.col = col; op.lazy = 1; }
void pushdown(int x) {
    if (!t[x].lazy) return;
    pushdown(t[x << 1], t[x].col), pushdown(t[x << 1 | 1], t[x].col);
    t[x].lazy = 0;
}

void build(int l, int r, int x = 1) {
    t[x] = { l, r, -1, 0 };
    if (l == r) return;
    int mid = l + r >> 1;
    build(l, mid, x << 1), build(mid + 1, r, x << 1 | 1);
}

void modify(int l, int r, int c, int x = 1) {
    if (l <= t[x].l && r >= t[x].r) {
        pushdown(t[x], c);
        return;
    }
    pushdown(x);
    int mid = t[x].l + t[x].r >> 1;
    if (l <= mid) modify(l, r, c, x << 1);
    if (r > mid) modify(l, r, c, x << 1 | 1);
}

int ask(int a, int x = 1) {
    if (t[x].l == a && t[x].r == a) return t[x].col;
    pushdown(x);
    int mid = t[x].l + t[x].r >> 1;
    return ask(a, x << 1 | (a > mid));
}


void init() {
    ind = 0;
    rep(i, n) edge[i].clear(), du[i] = 0;
    rep(i, n - 1) {
        int a, b; scanf("%d %d", &a, &b);
        edge[b].push_back(a);
        du[a]++;
    }
    
    rep(i, n) if (!du[i]) dfs(i);
    
    build(1, n);
}

int main()
{
    int T; cin >> T;
    rep(CASE, T) {
        printf("Case #%d:\n", CASE);
        cin >> n; init();
        cin >> m;
        while (m--) {
            char s[2]; scanf("%s", s);
            if (*s == 'T') {
                int x, c; scanf("%d %d", &x, &c);
                modify(BG[x], ED[x], c);
            }
            else {
                int x; scanf("%d", &x);
                printf("%d\n", ask(BG[x]));
            }
        }
    }
    return 0;
}

AC代码(树链剖分)

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
using namespace std;
typedef long long ll;
const int N = 5E4 + 10;
int n, m;
vector<int> edge[N];
int id[N], depth[N], sz[N], p[N], top[N], son[N], ind; 
//节点编号,  深度,     大小,   父亲,  链头,   重儿子
int du[N];

void dfs1(int x, int fa = 0, int level = 1) { 
    depth[x] = level, p[x] = fa, sz[x] = 1;
    for (auto& to : edge[x]) {
        if (to == fa) continue;
        dfs1(to, x, level + 1);
        sz[x] += sz[to];
        if (sz[son[x]] < sz[to]) son[x] = to;
    }
}
void dfs2(int x, int tp) {
    id[x] = ++ind, top[x] = tp;
    if (!son[x]) return;
    
    dfs2(son[x], tp);
    for (auto& to : edge[x]) {
        if (to == p[x] || to == son[x]) continue;
        dfs2(to, to);
    }
}


struct node {
    int l, r;
    int col;
    bool lazy;
}t[N << 2];

void pushdown(node& op, int col) { op.col = col; op.lazy = 1; }
void pushdown(int x) {
    if (!t[x].lazy) return;
    pushdown(t[x << 1], t[x].col), pushdown(t[x << 1 | 1], t[x].col);
    t[x].lazy = 0;
}

void build(int l, int r, int x = 1) {
    t[x] = { l, r, -1, 0 };
    if (l == r) return;
    int mid = l + r >> 1;
    build(l, mid, x << 1), build(mid + 1, r, x << 1 | 1);
}

void modify(int l, int r, int c, int x = 1) {
    if (l <= t[x].l && r >= t[x].r) {
        pushdown(t[x], c);
        return;
    }
    pushdown(x);
    int mid = t[x].l + t[x].r >> 1;
    if (l <= mid) modify(l, r, c, x << 1);
    if (r > mid) modify(l, r, c, x << 1 | 1);
}

int ask(int a, int x = 1) {
    if (t[x].l == a && t[x].r == a) return t[x].col;
    pushdown(x);
    int mid = t[x].l + t[x].r >> 1;
    return ask(a, x << 1 | (a > mid));
}


void init() {
    ind = 0;
    rep(i, n) edge[i].clear(), du[i] = son[i] = 0;
    rep(i, n - 1) {
        int a, b; scanf("%d %d", &a, &b);
        edge[b].push_back(a);
        du[a]++;
    }
    
    rep(i, n) if (!du[i]) dfs1(i), dfs2(i, i);
    
    build(1, n);
}

int main()
{
    int T; cin >> T;
    rep(CASE, T) {
        printf("Case #%d:\n", CASE);
        cin >> n; init();
        cin >> m;
        while (m--) {
            char s[2]; scanf("%s", s);
            if (*s == 'T') {
                int x, c; scanf("%d %d", &x, &c);
                modify(id[x], id[x] + sz[x] - 1, c);
            }
            else {
                int x; scanf("%d", &x);
                printf("%d\n", ask(id[x]));
            }
        }
    }
    return 0;
}

kuangbin线段树专题点这里!!!

END