码迷,mamicode.com
首页 > 其他好文 > 详细

BZOJ4557 [JLoi2016]侦察守卫 【树形dp】

时间:2018-05-08 19:23:15      阅读:118      评论:0      收藏:0      [点我收藏+]

标签:mes   最小   ons   stream   一个   algorithm   题目   class   付出   

题目链接

BZOJ4557

题解

orz
比较难的树形dp
不过想想也还好

看数据猜状态,一维是点,一维是D
那么就先设\(f[i][j]\)表示\(i\)所在子树已处理完毕,还能向上【或向任意方向】覆盖\(j\)层的最小代价
考虑转移,会发现子树间会相互影响,一个子树用\(f[s][j + 1]\)更新了\(f[i][j]\),其它的子树就完全没必要再用\(f[s‘][j + 1]\)去更新了,此时反而可以用\(f[i][j]\)来减少该子树付出的代价
所以我们设一个\(g[i][j]\)表示\(i\)为根的子树前\(j\)层待覆盖,\(j\)层以下已处理完毕的最小代价

\(f[i][j]\)包含\(f[i][j - 1]\),所以我们可以设状态\(f[i][j]\)\(j\)表示小于等于\(j\)
同样设\(g[i][j]\)中的\(j\)表示大于等于\(j\)

状态转移:枚举子树\(s\)
\[f[i][j] = min\{f[i][j] + g[s][j],f[s][j + 1] + g[i][j + 1]\}\]
\[g[i][j] += g[s][j - 1]\]

初始化就考虑\(i\)号点是不是一定被覆盖,就可以确定\(f[i][0]\)\(g[i][0]\)的初值

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define LL long long int
#define Redge(u) for (int k = h[u],to; k; k = ed[k].nxt)
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define BUG(s,n) for (int i = 1; i <= (n); i++) cout<<s[i]<<‘ ‘; puts("");
using namespace std;
const int maxn = 500005,INF = 1000000000;
inline int read(){
    int out = 0,flag = 1; char c = getchar();
    while (c < 48 || c > 57){if (c == ‘-‘) flag = -1; c = getchar();}
    while (c >= 48 && c <= 57){out = (out << 3) + (out << 1) + c - 48; c = getchar();}
    return out * flag;
}
int h[maxn],ne = 2;
struct EDGE{int to,nxt;}ed[maxn << 1];
inline void build(int u,int v){
    ed[ne] = (EDGE){v,h[u]}; h[u] = ne++;
    ed[ne] = (EDGE){u,h[v]}; h[v] = ne++;
}
int n,m,D,f[maxn][23],g[maxn][23],val[maxn],dan[maxn],fa[maxn];
int son[maxn],si;
void dfs(int u){
    for (int i = 1; i <= D; i++) f[u][i] = val[u];
    if (dan[u]) f[u][0] = g[u][0] = val[u];
    f[u][D + 1] = INF;
    Redge(u) if ((to = ed[k].to) != fa[u]){
        fa[to] = u; dfs(to);
        for (int i = 0; i <= D; i++)
            f[u][i] = min(f[u][i] + g[to][i],f[to][i + 1] + g[u][i + 1]);
        for (int i = D; i >= 0; i--)
            f[u][i] = min(f[u][i],f[u][i + 1]);
        g[u][0] = f[u][0];
        for (int i = 1; i <= D; i++)
            g[u][i] += g[to][i - 1];
        for (int i = 1; i <= D; i++)
            g[u][i] = min(g[u][i],g[u][i - 1]);
    }
}
int main(){
    n = read(); D = read();
    REP(i,n) val[i] = read();
    m = read();
    REP(i,m) dan[read()] = true;
    for (int i = 1; i < n; i++) build(read(),read());
    dfs(1);
    printf("%d\n",f[1][0]);
    return 0;
}

BZOJ4557 [JLoi2016]侦察守卫 【树形dp】

标签:mes   最小   ons   stream   一个   algorithm   题目   class   付出   

原文地址:https://www.cnblogs.com/Mychael/p/9009583.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!