洛谷【P2458】[SDOI2006]保安站岗题解树上DP-CSDN博客

题目描述

五一来临，某地下超市为了便于疏通和指挥密集的人员和车辆，以免造成超市内的混乱和拥挤，准备临时从外单位调用部分保安来维持交通秩序。

已知整个地下超市的所有通道呈一棵树的形状；某些通道之间可以互相望见。总经理要求所有通道的每个端点（树的顶点）都要有人全天候看守，在不同的通道端点安排保安所需的费用不同。

一个保安一旦站在某个通道的其中一个端点，那么他除了能看守住他所站的那个端点，也能看到这个通道的另一个端点，所以一个保安可能同时能看守住多个端点（树的结点），因此没有必要在每个通道的端点都安排保安。

编程任务：

请你帮助超市经理策划安排，在能看守全部通道端点的前提下，使得花费的经费最少。

输入输出格式

输入：

第\(1\)行 \(n\)，表示树中结点的数目。

第\(2\)行至第\(n+1\)行，每行描述每个通道端点的信息，依次为：该结点标号\(i（0<i<=n）\)，在该结点安置保安所需的经费\(k（<=10000）\)，该边的儿子数\(m\)，接下来\(m\)个数，分别是这个节点的\(m\)个儿子的标号\(r1，r2，...，rm\)。

对于一个\(n（0 < n <= 1500）\)个结点的树，结点标号在\(1\)到\(n\)之间，且标号不重复。

输出:

最少的经费。

如右图的输入数据示例

输入输出样例

输入样例#1：

6
1 30 3 2 3 4
2 16 2 5 6
3 5 0
4 4 0
5 11 0
6 5 0

输出样例#1：

说明

样例说明：在结点\(2，3，4\)安置\(3\)个保安能看守所有的\(6\)个结点，需要的经费最小：\(25\)

根据题目描述很容易想到是树上\(DP\)，但是有很多需要考虑到的东西：

首先容易被坑的一点：

很可能会直接想到两种状态：当前点占用和不占用。
在这种情况下会很容易就作出如下的判断
- \(F[u][1]+=min\{F[v][0],F[v][1]\};\)（当前点占用的转移）
- \(F[u][0]+=F[v][1];\)（当前点不占用的转移）

然后就会第一时间\(w\)\(\bar a\)掉这个题目。

为什么呢？因为这个转移中，我们忽略了子节点对父节点的影响，子节点的选中也可以控制父节点，所以我们的状态应该有三个：

当前点被占用（子节点选啥都可以，直接取最小值）
当前点没有被占用（父节点占用并控制该点，所以可以把子节点里面已经独立可用的都挖过来）
当前点没有被占用，但是子节点中存在有已经被占用的点
- 这个转移相对不是很好考虑，怎么确认子节点中谁被占用呢？
- 我们先考虑不管子节点有没有被占用的，直接先取两种转移状态最小值临时存起来
- 然后一一比较，看哪一个子节点变成占用态消耗最少，答案取min
- 复杂度\(O(n)\)

实际上算是树上\(DP\)的一种常用套路吧，不过本蒻以前没有掌握，被这个题目好好教育了一下。

Code：

#include<cstdio>
#include<cstring>
#include<iostream>
#define MAXN 1510
#define INF 0x3f3f3f3f
using namespace std;
int cnt,head[MAXN],deep[MAXN];
int n,vis[MAXN],arr[MAXN],size[MAXN],f[MAXN][4];
struct edge{
    int nxt;
    int to;
}e[MAXN<<1];
void dfs(int u,int fa){
    f[u][1]=arr[u];//预先处理被占用的情况
    int sum=0;
    for(int i=head[u];i;i=e[i].nxt){
        int v=e[i].to;
        if(v!=fa){
            //对子树的处理
            dfs(v,u);
            f[u][1]+=min(f[v][1],min(f[v][2],f[v][3]));
            sum+=min(f[v][1],f[v][2]);
        }
    }
    f[u][3]=sum,f[u][2]=INF;
    for(int i=head[u];i;i=e[i].nxt){
        int v=e[i].to;
        if(v!=fa){
            f[u][2]=min(f[u][2],sum-min(f[v][1],f[v][2])+f[v][1]);
        }
    }
//  printf("f[%d][1]=%d f[%d][2]=%d f[%d][3]=%d\n",u,f[u][1],u,f[u][2],u,f[u][3]);
    return; 
}

inline void add(int from,int to){
    e[++cnt].nxt=head[from];
    e[cnt].to=to;
    head[from]=cnt;
}
 
int main(){
    int u,v,m;
    scanf("%d",&n);
    for(int i=1;i<=n;++i){
        scanf("%d",&u);
        scanf("%d%d",&arr[u],&m);
        for(int j=1;j<=m;++j){
            scanf("%d",&v);
            add(u,v);
            add(v,u);
        }
    }//arr记录价格 
    //f[i][1]->本节点占用时子树最小花费
    //f[i][2]->本节点不占用但被控制时子树最小花费
    //f[i][3]->本节点完全不受控制时子树最小花费
    dfs(1,0);
//  for(int i=1;i<=n;++i){
//      printf("size %d = %d\n",i,size[i]);
//  }
//  size保存子树大小 
    printf("%d\n",min(f[1][1],f[1][2]));
    return 0;
}

转载于:https://www.cnblogs.com/maomao9173/p/9858665.html