[ZJOI2008]树的统计 题解

[ZJOI2008]树的统计 题解

题目地址:洛谷:【P2590】[ZJOI2008]树的统计 – 洛谷、BZOJ:Problem 1036. — [ZJOI2008]树的统计Count

题目描述

一棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w。
我们将以下面的形式来要求你对这棵树完成一些操作:
I. CHANGE u t : 把结点u的权值改为t
II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值
III. QSUM u v: 询问从点u到点v的路径上的节点的权值和
注意:从点u到点v的路径上的节点包括u和v本身

输入输出格式

输入格式:
输入文件的第一行为一个整数n,表示节点的个数。
接下来n – 1行,每行2个整数a和b,表示节点a和节点b之间有一条边相连。
接下来一行n个整数,第i个整数wi表示节点i的权值。
接下来1行,为一个整数q,表示操作的总数。
接下来q行,每行一个操作,以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。

输出格式:
对于每个“QMAX”或者“QSUM”的操作,每行输出一个整数表示要求输出的结果。

输入输出样例

输入样例#1:

4
1 2
2 3
4 1
4 2 1 3
12
QMAX 3 4
QMAX 3 3
QMAX 3 2
QMAX 2 3
QSUM 3 4
QSUM 2 1
CHANGE 1 5
QMAX 3 4
CHANGE 3 6
QMAX 3 4
QMAX 2 4
QSUM 3 4

输出样例#1:

4
1
2
2
10
6
5
6
5
16

说明

对于100%的数据,保证1<=n<=30000,0<=q<=200000;中途操作中保证每个节点的权值w在-30000到30000之间。

题解

比洛谷树剖模板还简单。就是个裸树剖。

代码

// Code by KSkun, 2018/4
#include <cstdio>

#include <algorithm>
#include <vector>

typedef long long LL;

inline char fgc() {
    static char buf[100000], *p1 = buf, *p2 = buf;
    return p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 100000, stdin), p1 == p2) ? EOF 
        : *p1++;
}

inline LL readint() {
    register LL res = 0, neg = 1;
    char c = fgc();
    while(c < '0' || c > '9') {
        if(c == '-') neg = -1;
        c = fgc();
    }
    while(c >= '0' && c <= '9') {
        res = res * 10 + c - '0';
        c = fgc();
    }
    return res * neg;
}

inline bool isop(char c) {
    return c == 'X' || c == 'S' || c == 'C';
}

inline char readop() {
    char c;
    while(!isop(c = fgc()));
    return c;
}

const int MAXN = 30005, INF = 1e9;

int n, q, dfn[MAXN], ptn[MAXN], w[MAXN];
std::vector<int> gra[MAXN];

#define lch o << 1
#define rch (o << 1) | 1
#define mid ((l + r) >> 1)

LL val[MAXN << 2], mx[MAXN << 2];

inline void merge(int o) {
    val[o] = val[lch] + val[rch];
    mx[o] = std::max(mx[lch], mx[rch]);
}

inline void build(int o, int l, int r) {
    if(l == r) {
        val[o] = mx[o] = w[ptn[l]];
        return;
    }
    build(lch, l, mid);
    build(rch, mid + 1, r);
    merge(o);
}

inline void modify(int o, int l, int r, int x, LL v) {
    if(l == r) {
        val[o] = mx[o] = v;
        return;
    }
    if(x <= mid) modify(lch, l, mid, x, v);
    else modify(rch, mid + 1, r, x, v);
    merge(o);
}

inline LL querys(int o, int l, int r, int ll, int rr) {
    if(l >= ll && r <= rr) {
        return val[o];
    }
    LL res = 0;
    if(ll <= mid) res += querys(lch, l, mid, ll, rr);
    if(rr > mid) res += querys(rch, mid + 1, r, ll, rr);
    return res;
}
inline LL querym(int o, int l, int r, int ll, int rr) {
    if(l >= ll && r <= rr) {
        return mx[o];
    }
    LL res = -INF;
    if(ll <= mid) res = std::max(res, querym(lch, l, mid, ll, rr));
    if(rr > mid) res = std::max(res, querym(rch, mid + 1, r, ll, rr));
    return res;
}

int fa[MAXN], siz[MAXN], dep[MAXN], top[MAXN], son[MAXN], clk;

inline void dfs1(int u) {
    siz[u] = 1;
    for(int i = 0; i < gra[u].size(); i++) {
        int v = gra[u][i];
        if(v == fa[u]) continue;
        fa[v] = u;
        dep[v] = dep[u] + 1;
        dfs1(v);
        siz[u] += siz[v];
        if(siz[v] > siz[son[u]]) son[u] = v;
    }
}

inline void dfs2(int u, int tp) {
    dfn[u] = ++clk;
    ptn[dfn[u]] = u;
    top[u] = tp;
    if(son[u]) {
        dfs2(son[u], tp);
    }
    for(int i = 0; i < gra[u].size(); i++) {
        int v = gra[u][i];
        if(v == fa[u] || v == son[u]) continue;
        dfs2(v, v);
    }
}

inline LL querys(int x, int y) {
    int tx = top[x], ty = top[y];
    LL res = 0;
    while(tx != ty) {
        if(dep[tx] > dep[ty]) {
            std::swap(x, y);
            std::swap(tx, ty);
        }
        res += querys(1, 1, n, dfn[ty], dfn[y]);
        y = fa[ty];
        ty = top[y];
    }
    if(dep[x] > dep[y]) {
        std::swap(x, y);
    }
    res += querys(1, 1, n, dfn[x], dfn[y]);
    return res;
}

inline LL querym(int x, int y) {
    int tx = top[x], ty = top[y];
    LL res = -INF;
    while(tx != ty) {
        if(dep[tx] > dep[ty]) {
            std::swap(x, y);
            std::swap(tx, ty);
        }
        res = std::max(res, querym(1, 1, n, dfn[ty], dfn[y]));
        y = fa[ty];
        ty = top[y];
    }
    if(dep[x] > dep[y]) {
        std::swap(x, y);
    }
    res = std::max(res, querym(1, 1, n, dfn[x], dfn[y]));
    return res;
}

inline void modify(int x, LL z) {
    modify(1, 1, n, dfn[x], z);
}

int x, y;
char op;

int main() {
    n = readint();
    for(int i = 1; i < n; i++) {
        x = readint(); y = readint();
        gra[x].push_back(y);
        gra[y].push_back(x);
    }
    for(int i = 1; i <= n; i++) w[i] = readint();
    dfs1(1);
    dfs2(1, 1);
    build(1, 1, n);
    q = readint();
    while(q--) {
        op = readop(); x = readint(); y = readint();
        if(op == 'X') {
            printf("%lld\n", querym(x, y));
        } else if(op == 'S') {
            printf("%lld\n", querys(x, y));
        } else {
            modify(x, y);
        }
    }
    return 0;
}


发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注

This site is protected by reCAPTCHA and the Google Privacy Policy and Terms of Service apply.

此站点使用Akismet来减少垃圾评论。了解我们如何处理您的评论数据