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

来源:互联网 发布:淘宝旺旺位置设置 编辑:程序博客网 时间:2024/04/30 16:00

题目传送门
题意解析:题目就是告诉了你一棵树,再告诉你在一个点放置一个覆盖点的代价,然后每在一个点放置覆盖点,离它距离小于等于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 20000000using 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纪录了,没法截图。

原创粉丝点击