bzoj5210 最大连通子块和动态 DP + 堆

最新推荐文章于 2024-09-24 07:29:36 发布

weixin_30455023

最新推荐文章于 2024-09-24 07:29:36 发布

阅读量103

点赞数

文章标签： php

原文链接：http://www.cnblogs.com/hankeke/p/bzoj5210.html

版权

题目传送门

https://lydsy.com/JudgeOnline/problem.php?id=5210

题解

令 \(dp[x][0]\) 表示以 \(x\) 为根的子树中的包含 \(x\) 的连通块的点权和最大值，\(dp[x][1]\) 为 \(x\) 的所有孩子中的子树的连通块点权和最大值。

于是有这样的式子：
\[ dp[x][0] = \max(w_x+\sum_{y\in child_x} dp[y][0], 0)\\dp[x][1] = \max_{y\in child_x} \max(dp[y][0], dp[y][1]) \]
（其实似乎把 \(dp[x][1]\) 算上 \(dp[x][0]\) 的内容也没什么问题，但是我一开始就这样想的不想改了）

现在把这个 \(dp\) 动态化。令树剖以后，\(x\) 的重儿子为 \(son_x\)，\(x\) 的轻儿子的集合为 \(light_x\)。令 \(h[x][0] = \sum\limits_{y \in light_x} dp[y][0]\)，\(h[x][1] = \max\limits_{y\in light_x} max(dp[y][0], dp[y][1])\)。

于是上面的方程可以改写为
\[ dp[x][0] = \max(w_x+h[x][0] + dp[son_x][0], 0)\\dp[x][1] = \max(h[x][1], dp[son_x][1], dp[son_x][0]) \]
于是矩阵（矩乘是把乘法看成加法，加法看成取 \(\max\)）就长这样：
\[ \begin{bmatrix}h[x][0] + w_x & -\infty & 0\\ 0 & 0 & h[x][1] \\ -\infty & -\infty & 0\end{bmatrix} \times \begin{bmatrix} dp[son_x][0] \\ dp[son_x][1] \\ 0 \end{bmatrix} = \begin{bmatrix} dp[x][0] \\ dp[x][1] \\ 0 \end{bmatrix} \]
然后就套一个动态 DP 的板子就可以了。

有一个需要注意的地方就是因为 \(dp[x][1]\) 是取 \(\max\) 的，所以在更新 \(dp[x][1]\) 的时候需要用一个可删除堆来维护最大值。

昨天这道题因为一个 ll 没有写调了几个小时。

单次询问是 \(O(\log n)\) 的，单次修改是 \(O(\log^2 n)\)。

#include<bits/stdc++.h>

#define fec(i, x, y) (int i = head[x], y = g[i].to; i; i = g[i].ne, y = g[i].to)
#define dbg(...) fprintf(stderr, __VA_ARGS__)
#define File(x) freopen(#x".in", "r", stdin), freopen(#x".out", "w", stdout)
#define fi first
#define se second
#define pb push_back

template<typename A, typename B> inline char smax(A &a, const B &b) {return a < b ? a = b, 1 : 0;}
template<typename A, typename B> inline char smin(A &a, const B &b) {return b < a ? a = b, 1 : 0;}

typedef long long ll; typedef unsigned long long ull; typedef std::pair<int, int> pii;

template<typename I> inline void read(I &x) {
    int f = 0, c;
    while (!isdigit(c = getchar())) c == '-' ? f = 1 : 0;
    x = c & 15;
    while (isdigit(c = getchar())) x = (x << 1) + (x << 3) + (c & 15);
    f ? x = -x : 0;
}

#define lc o << 1
#define rc o << 1 | 1

const int N = 200000 + 7;
const ll INF = 0x3f3f3f3f3f3f3f3f;

int n, m, dfc;
int a[N];
int dep[N], f[N], siz[N], son[N], dfn[N], pre[N], top[N], bot[N];
ll dp[N][2], h[N][2], hh[N][2];

struct Heap {
    std::priority_queue<ll> q, p;
    inline void sync() { while (!q.empty() && !p.empty() && q.top() == p.top()) q.pop(), p.pop(); }
    inline void push(ll x) { q.push(x); sync(); }
    inline void del(ll x) { p.push(x), sync(); }
    inline ll top() { sync(); return q.top(); }
    inline bool empty() { sync(); return q.empty(); }
    inline int size() { sync(); return q.size() - p.size·(); }
} q[N];

struct Edge { int to, ne; } g[N << 1]; int head[N], tot;
inline void addedge(int x, int y) { g[++tot].to = y, g[tot].ne = head[x], head[x] = tot; }
inline void adde(int x, int y) { addedge(x, y), addedge(y, x); }

struct Matrix {
    ll a[3][3];
    inline Matrix() { memset(a, 0, sizeof(a)); }
    inline Matrix operator * (const Matrix &b) {
        Matrix c;
        c.a[0][0] = std::max(std::max(a[0][0] + b.a[0][0], a[0][1] + b.a[1][0]), a[0][2] + b.a[2][0]);
        c.a[0][1] = std::max(std::max(a[0][0] + b.a[0][1], a[0][1] + b.a[1][1]), a[0][2] + b.a[2][1]);
        c.a[0][2] = std::max(std::max(a[0][0] + b.a[0][2], a[0][1] + b.a[1][2]), a[0][2] + b.a[2][2]);
        c.a[1][0] = std::max(std::max(a[1][0] + b.a[0][0], a[1][1] + b.a[1][0]), a[1][2] + b.a[2][0]);
        c.a[1][1] = std::max(std::max(a[1][0] + b.a[0][1], a[1][1] + b.a[1][1]), a[1][2] + b.a[2][1]);
        c.a[1][2] = std::max(std::max(a[1][0] + b.a[0][2], a[1][1] + b.a[1][2]), a[1][2] + b.a[2][2]);
        c.a[2][0] = std::max(std::max(a[2][0] + b.a[0][0], a[2][1] + b.a[1][0]), a[2][2] + b.a[2][0]);
        c.a[2][1] = std::max(std::max(a[2][0] + b.a[0][1], a[2][1] + b.a[1][1]), a[2][2] + b.a[2][1]);
        c.a[2][2] = std::max(std::max(a[2][0] + b.a[0][2], a[2][1] + b.a[1][2]), a[2][2] + b.a[2][2]);
        return c;
    }
} t[N << 2];

inline void dfs1(int x, int fa = 0) {
    dep[x] = dep[fa] + 1, f[x] = fa, siz[x] = 1;
    for fec(i, x, y) if (y != fa) dfs1(y, x), siz[x] += siz[y], siz[y] > siz[son[x]] && (son[x] = y);
}
inline void dfs2(int x, int pa) {
    top[x] = pa, dfn[x] = ++dfc, pre[dfc] = x;
    if (!son[x]) return (void)(bot[x] = x);
    dfs2(son[x], pa), bot[x] = bot[son[x]];
    for fec(i, x, y) if (y != f[x] && y != son[x]) dfs2(y, y);
}
inline void dfs3(int x, int fa = 0) {
    dp[x][0] = a[x];
    for fec(i, x, y) if (y != fa) {
        dfs3(y, x);
        dp[x][0] += dp[y][0];
        smax(dp[x][1], std::max(dp[y][1], dp[y][0]));
        if (y != son[x]) h[x][0] += dp[y][0], smax(h[x][1], std::max(dp[y][1], dp[y][0])), q[x].push(std::max(dp[y][1], dp[y][0]));
    }
    smax(dp[x][0], 0);
}

inline void build(int o, int L, int R) {
    if (L == R) {
        int x = pre[L];
        t[o].a[0][0] = h[x][0] + a[x], t[o].a[0][1] = -INF, t[o].a[0][2] = 0;
        t[o].a[1][0] = 0, t[o].a[1][1] = 0, t[o].a[1][2] = h[x][1];
        t[o].a[2][0] = -INF, t[o].a[2][1] = -INF, t[o].a[2][2] = 0;
        return;
    }
    int M = (L + R) >> 1;
    build(lc, L, M), build(rc, M + 1, R);
    t[o] = t[lc] * t[rc];
}
inline void qadd(int o, int L, int R, int x) {
    if (L == R) {
        int x = pre[L];
        t[o].a[0][0] = h[x][0] + a[x], t[o].a[0][1] = -INF, t[o].a[0][2] = 0;
        t[o].a[1][0] = 0, t[o].a[1][1] = 0, t[o].a[1][2] = h[x][1];
        t[o].a[2][0] = -INF, t[o].a[2][1] = -INF, t[o].a[2][2] = 0;
        return;
    }
    int M = (L + R) >> 1;
    if (x <= M) qadd(lc, L, M, x);
    else qadd(rc, M + 1, R, x);
    t[o] = t[lc] * t[rc];
}
inline Matrix qsum(int o, int L, int R, int l, int r) {
    if (l <= L && R <= r) return t[o];
    int M = (L + R) >> 1;
    if (r <= M) return qsum(lc, L, M, l, r);
    if (l > M) return qsum(rc, M + 1, R, l, r);
    return qsum(lc, L, M, l, r) * qsum(rc, M + 1, R, l, r);
}

inline Matrix qry(int x) { return qsum(1, 1, n, dfn[x], dfn[bot[x]]) * Matrix(); }
inline void upd(int x, int k) {
    a[x] = k;
    while (top[x] != 1) {
        Matrix tmp1 = qry(top[x]);
        qadd(1, 1, n, dfn[x]);
        Matrix tmp2 = qry(top[x]);
        int fa = f[top[x]];
        h[fa][0] += tmp2.a[0][0] - tmp1.a[0][0];
        q[fa].del(std::max(tmp1.a[0][0], tmp1.a[1][0])), q[fa].push(std::max(tmp2.a[0][0], tmp2.a[1][0])), h[fa][1] = q[fa].top();
        x = fa;
    }
    qadd(1, 1, n, dfn[x]);
}

inline void work() {
    dfs1(1), dfs2(1, 1), dfs3(1), build(1, 1, n);
    Matrix tmp;
    while (m--) {
        static char opt[5];
        int x, y;
        scanf("%s", opt);
        if (*opt == 'Q') read(x), tmp = qry(x), printf("%lld\n", std::max(tmp.a[0][0], tmp.a[1][0]));
        else read(x), read(y), upd(x, y);
    }
}

inline void init() {
    read(n), read(m);
    for (int i = 1; i <= n; ++i) read(a[i]);
    int x, y;
    for (int i = 1; i < n; ++i) read(x), read(y), adde(x, y);
}

int main() {
#ifdef hzhkk
    freopen("hkk.in", "r", stdin);
#endif
    init();
    work();
    fclose(stdin), fclose(stdout);
    return 0;
}

转载于:https://www.cnblogs.com/hankeke/p/bzoj5210.html