[bzoj-4557][JLoi2016]侦察守卫 题解

题目传送门
题意解析：题目就是告诉了你一棵树，再告诉你在一个点放置一个覆盖点的代价，然后每在一个点放置覆盖点，离它距离小于等于d的都会被覆盖，问覆盖要求点的最小代价。

---

My opinion:我这题曾经听dalao讲过，然而后来我做的时候还是不会，只是知道这题目是树形dp，所以当场做的时候就翻车了。而且这题是求要求的覆盖点的，而我会的只是覆盖全部的点，后来回去仔细想，才发现覆盖要求点和全部点的区别就是初值的定义。我们只需要在要求的点上给初值就好了。不过因为如果在这个点上放置的话，还可能跟它的祖先有关，所以我们需要用两个dp数组表示不同的情况。
总结：
1、输入。
2、dfs时树迪（树形dp）。
f[i][j]表示覆盖i子树中与i距离超过j的要求点的最小代价。
g[i][j]表示以i为根的子树已经全部覆盖后，还能向上j步的最小代价。
3、输出。

---

代码：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define rep(i,a,n) for (int i=a;i<=n;i++)
#define per(i,a,n) for (int i=a;i>=n;i--)
#define Clear(a,x) memset(a,x,sizeof(a))
#define ll long long
#define eps 1e-8
#define INF 20000000
using namespace std;
int read(){
    int x=0,f=1;
    char ch=getchar();
    while (ch<'0'||ch>'9') f=ch=='-'?-1:f,ch=getchar();
    while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
const int maxn=500005;
int f[maxn][25],g[maxn][25],w[maxn];
bool flag[maxn];
int vet[maxn<<1],Next[maxn<<1],head[maxn];
int n,m,d,len;
void add(int u,int v){
    vet[++len]=v;
    Next[len]=head[u];
    head[u]=len;
}
void dfs(int u,int fa){
    if (flag[u]) f[u][0]=g[u][0]=w[u];
    rep(i,1,d) g[u][i]=w[u];
    g[u][d+1]=INF;
    for (int e=head[u];e;e=Next[e]){
        int v=vet[e];
        if (v==fa) continue;
        dfs(v,u);
        per(j,d,0) g[u][j]=min(g[u][j]+f[v][j],g[v][j+1]+f[u][j+1]);
        per(j,d,0) g[u][j]=min(g[u][j],g[u][j+1]);
        f[u][0]=g[u][0];
        rep(j,1,d+1) f[u][j]+=f[v][j-1];
        rep(j,1,d+1) f[u][j]=min(f[u][j],f[u][j-1]);
    }
}
int main(){
    n=read(),d=read();
    rep(i,1,n) w[i]=read();
    m=read();
    rep(i,1,m) flag[read()]=1;
    rep(i,1,n-1){
        int u=read(),v=read();
        add(u,v),add(v,u);
    }
    dfs(1,0);
    printf("%d\n",f[1][0]);
    return 0;
}

不能附上AC纪录了，没法截图。