【bzoj4557】【JLoi2016】【树规】侦察守卫

侦察守卫

Time Limit: 20 Sec Memory Limit: 256 MB
Submit: 184 Solved: 123

Description

小R和B神正在玩一款游戏。这款游戏的地图由N个点和N-1条无向边组成，每条无向边连接两个点，且地图是连通的
。换句话说，游戏的地图是一棵有N个节点的树。游戏中有一种道具叫做侦查守卫，当一名玩家在一个点上放置侦
查守卫后，它可以监视这个点以及与这个点的距离在D以内的所有点。这里两个点之间的距离定义为它们在树上的
距离，也就是两个点之间唯一的简单路径上所经过边的条数。在一个点上放置侦查守卫需要付出一定的代价，在不
同点放置守卫的代价可能不同。现在小R知道了所有B神可能会出现的位置，请你计算监视所有这些位置的最小代价
。

Input

第一行包含两个正整数N和D，分别表示地图上的点数和侦查守卫的视野范围。约定地图上的点用1到N的整数编号。
第二行N个正整数，第i个正整数表示在编号为i的点放置侦查守卫的代价Wi。保证Wi≤1000。第三行一个正整数M，
表示B神可能出现的点的数量。保证M≤N。第四行M个正整数，分别表示每个B神可能出现的点的编号，从小到大不
重复地给出。接下来N–1行，每行包含两个正整数U,V，表示在编号为U的点和编号为V的点之间有一条无向边。N<=
500000,D<=20

Output

仅一行一个整数，表示监视所有B神可能出现的点所需要的最小代价

Sample Input

12 2
8 9 12 6 1 1 5 1 4 8 10 6
10
1 2 3 5 6 7 8 9 10 11
1 3
2 3
3 4
4 5
4 6
4 7
7 8
8 9
9 10
10 11
11 12

Sample Output

10

这道树规转移比较复杂，要想清楚，以至于我再也不想写DP
dp这里有两个数组dp1[i][j]表示i节点被覆盖同时可以向上覆盖j个节点，dp2表示i节点以及以下的j个节点都要被覆盖
然后注意dp1[x]0]=dp2[x][0]
转移过程见程序，就不详细写了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<set>
#include<map>
#include<queue>
#include<algorithm>
#include<vector>
#include<cstdlib>
#include<cmath>
#include<ctime>
#include<stack>
#define INF 2100000000
#define ll long long
#define clr(x)  memset(x,0,sizeof(x))
#define maxclr(x)  memset(x,63,sizeof(x))

using namespace std;

inline int read()
{
    char c;
    int ret=0;
    while(!(c>='0'&&c<='9'))
        c=getchar();
    while(c>='0'&&c<='9')
    {
        ret=(c-'0')+(ret<<1)+(ret<<3);
        c=getchar();
    }
    return ret;
}

#define M 500005
int n,d,w[M],m;
int first[M],next[M<<1],to[M<<1],t;
int dp1[M][21],dp2[M][21],b[M][21];

void addedge(int s,int v)
{
    next[++t]=first[s];
    first[s]=t;
    to[t]=v;
}

void dfs(int x,int fa)
{
    int flag=0;
    for(int i=first[x];i;i=next[i])
    {
        int v=to[i];
        if(v==fa)continue;
        dfs(v,x);
        flag=1;
        for(int j=1;j<=d;j++)
            b[x][j]|=b[v][j-1];//上传需要覆盖的位置 
    }
    if(!flag)//叶子节点 
    {
         dp1[x][0]=dp2[x][0]=w[x]*b[x][0];
         for(int i=1;i<=d;i++)
             dp1[x][i]=w[x],dp2[x][i]=0;
         return ;
    }

    for(int j=1;j<=d;j++)//更新dp2 
        for(int i=first[x];i;i=next[i])
            if(to[i]!=fa)
                dp2[x][j]+=dp2[to[i]][j-1];

    for(int i=first[x];i;i=next[i])//选取当前值
        if(to[i]!=fa)
            dp1[x][d]+=dp2[to[i]][d];
    dp1[x][d]+=w[x];

    for(int j=d-1;j>=0;j--)//更新dp1 
    {
        int sum=0;
        for(int i=first[x];i;i=next[i])
            if(to[i]!=fa)
                sum+=dp2[to[i]][j];
        dp1[x][j]=dp1[x][j+1];
        for(int i=first[x];i;i=next[i])
        {
            int v=to[i];
            if(v==fa)continue;
            dp1[x][j]=min(dp1[x][j],dp1[v][j+1]+sum-dp2[v][j]);
        }
    }

    dp2[x][0]=dp1[x][0];//dp2可以舍弃步数 
    for(int i=1;i<=d;i++)
        dp2[x][i]=min(dp2[x][i],dp2[x][i-1]);

    for(int i=d-1;i>=0;i--)//如果不用覆盖这个点 就要反向更新 
        if(!b[x][i])dp2[x][i]=min(dp2[x][i],dp2[x][i+1]);
    dp1[x][0]=dp2[x][0];    
}

int main()
{
    freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);
    n=read();d=read();
    for(int i=1;i<=n;i++)
        w[i]=read();
    m=read();
    for(int i=1;i<=m;i++)
    {
        int x=read();
        b[x][0]=1;
    }
    for(int i=1;i<n;i++)
    {
        int a=read(),b=read();
        addedge(a,b);
        addedge(b,a);
    }
    dfs(1,0);
    printf("%d\n",dp2[1][0]);
}

大概就是这个样子，如果有什么问题，或错误，请在评论区提出，谢谢。