Luogu 3267 [JLOI2016/SHOI2016]侦察守卫

以后要记得复习鸭

BZOJ 4557

大佬的博客

状态十分好想，设$f_{x, i}$表示以覆盖完$x$为根的子树后还能向上覆盖$i$层的最小代价，$g_{x, i}$表示以$x$为根的子树下深度为$i$还没有被覆盖的最小代价。

那么对于每一个关键点，有初态：   $f_{x,0} = g_{x, 0} = val_x$。

对于不是关键点的点，有：$f_{x, i} = val_x$   $0 \leq i \leq d$   $f_{x, d + 1} = inf$。

然后就不会了

感觉关键是维护$f_x$和$g_x$的单调性，因为$f_{x, i}$一定不会大于$f_{x, i + 1}$，而$g_{x, i}$一定不会大于$g_{x, i - 1}$。

我们考虑对$x$的所有儿子$y$分开考虑，对于每一个$y$，有方程

　　　　$f_{x, i} = min(f_{x, i} + g_{y, i}, g_{x, i + 1} + f_{y, i + 1})$。

前者表示在之前的计算中已经取了一个子结点能覆盖到$x$，现在这个结点只需要取$g_{y, i}$即可，后者表示取这个结点的$f_{y, i + 1}$，而剩下的代价最小为$g_{x, i + 1}$。

记得从上到下枚举$i$。

然后再扫一遍维护$f_{x}$的单调性。

接着考虑计算$g$，有$g_{x, 0} = f_{x, 0}$。

对于每一个$y$，有$g_{x, i} += g_{y, i - 1}$。

然后从下到上扫一遍维护$g_x$的单调性。

时间复杂度$O(nd)$。

Code：

#include <cstdio>
#include <cstring>
using namespace std;

const int N = 5e5 + ;
const int M = ;
const int inf =  << ;

int n, m, d, a[N], tot = , head[N], f[N][M], g[N][M];
bool flag[N];

struct Edge {
    int to, nxt;
} e[N << ];

inline void add(int from, int to) {
    e[++tot].to = to;
    e[tot].nxt = head[from];
    head[from] = tot;
}

inline void read(int &X) {
    X = ; char ch = ; int op = ;
    for(; ch > '' || ch < ''; ch = getchar())
        if(ch == '-') op = -;
    for(; ch >= '' && ch <= ''; ch = getchar())
        X = (X << ) + (X << ) + ch - ;
    X *= op;
}

inline int min(int x, int y) {
    return x > y ? y : x;
}

inline void chkMin(int &x, int y) {
    if(y < x) x = y;
}

void dfs(int x, int fat) {
    if(flag[x]) g[x][] = f[x][] = a[x];
    for(int i = ; i <= d; i++) f[x][i] = a[x];
    f[x][d + ] = inf;

    for(int i = head[x]; i; i = e[i].nxt) {
        int y = e[i].to;
        if(y == fat) continue;

        dfs(y, x);

        for(int j = d; j >= ; j--)
            f[x][j] = min(f[x][j] + g[y][j], g[x][j + ] + f[y][j + ]);
        for(int j = d; j >= ; j--) chkMin(f[x][j], f[x][j + ]);

        g[x][] = f[x][];
        for(int j = ; j <= d; j++) g[x][j] += g[y][j - ];
        for(int j = ; j <= d; j++) chkMin(g[x][j], g[x][j - ]);
    }
}

int main() {
    read(n), read(d);
    for(int i = ; i <= n; i++) read(a[i]);
    read(m);
    for(int x, i = ; i <= m; i++) {
        read(x);
        flag[x] = ;
    }
    for(int x, y, i = ; i < n; i++) {
        read(x), read(y);
        add(x, y), add(y, x);
    }

    dfs(, );

    printf("%d\n", f[][]);
    return ;
}