UVA - 12093 Protecting Zonk（树形DP经典难题）

传送门

---

时隔两个多月前来补题，这题真的有点难搞（吐槽= =）

Reference：
https://www.luogu.com.cn/problem/solution/UVA12093题解第二篇
https://www.cnblogs.com/djh0709/p/9606749.html

首先看到题目，不难发现对于某个节点，两种装置的影响如下图所示：

其中分别用英文缩写代表了祖父，父亲，兄弟，当前节点$u$，儿子，孙子。这已经是影响的最大范围了

但是我们考虑到编程难度，树形DP不就是在树上$dfs$然后利用回溯更新状态，显然对于一个节点我们最多只知道父节点和子节点，而且当前节点状态的更新也只能由子节点更新，那么就必须找到正确的覆盖面完全的状态转移方程，那么我们就将放置两种装置等效于影响的边数。由于本人只是蒟蒻，只能从已知题解中理解了，但是我感觉我的理解应该有百分之九十多了，完胜大多数水的题解

先列出状态转移方程，设：

• $d(u,0)$表示以点$u$为根的子树下的边全部被覆盖，且没有向u节点上方覆盖。换句话说也就是$u$不安装装置，而$u$和儿子的连边被儿子覆盖，而且儿子不能装$B$，那么儿子要么装$A$，要么就是孙子装了$B$使得$u$和儿子的连边被覆盖，那么状态的转移显而易见： d ( u , 0 ) = s u m { m i n ( d ( v , 0 ) , d ( v , 1 ) ) } d(u,0)=sum\{min(d(v,0),d(v,1))\} d(u,0)=sum{min(d(v,0),d(v,1))}
• $d(u,1)$表示以点$u$为根的子树下的边全被覆盖，而且向上覆盖了一条边，其实也就是覆盖了$u$和其父亲相连的边。换句话说，如果想向上覆盖一条边，要么$u$装了$A$，要么儿子装了$B$，因此需要比较是在儿子安装$B$划算还是在$u$安装$A$划算。但是不要忘了$u$的子树全被覆盖这个条件，可以从儿子的前三个状态转移，为什么不从第四个转移还是不难想到的（留给读者思考）。因此状态转移方程为 d ( u , 1 ) = s u m { m i n ( d ( v , 0 ) , d ( v , 1 ) , d ( v , 2 ) ) + m i n ( c 1 , m i n ( d ( v , 2 ) − m i n ( d ( v , 0 ) , d ( v , 1 ) , d ( v , 2 ) ) ) } d(u,1)= sum\{ min( d(v,0),d(v,1),d(v,2)) +min( c1, min( d(v,2)-min( d(v,0),d(v,1),d(v,2) ) )\} d(u,1)=sum{min(d(v,0),d(v,1),d(v,2))+min(c1,min(d(v,2)−min(d(v,0),d(v,1),d(v,2)))}，这里之所以后面需要减去一部分，是因为如果一个儿子装了$B$那么就不需要从其他两个状态更新了，也就是说这里多加了装$B$的那个儿子的三种转移状态的其中一种，因此要减去
• $d(u,2)$表示以点$u$为根的子树下的边全被覆盖，而且向上覆盖了两条边，其实也就是一条覆盖了$u$和其父亲相连的边，另外一条可以覆盖父亲和祖父的，也可以覆盖父亲和兄弟的。那么显然儿子不管是什么状态都不能向上覆盖两条边，这个时候只能是在$u$装$B$，但是$u$装了$B$之后儿子和孙子都可以不装，那么也就是能从儿子的四个状态转移而来
• $d(u,3)$表示以点$u$的儿子为根的子树下的边全被覆盖，而且$u$和儿子的连边已经被父亲覆盖。这个状态相当于是第三个状态的补充，因为两个相邻节点都不装的状态是没有的。虽然$u$和儿子的边已经被覆盖，但是对于$u$的儿子来说还是能从前三种状态转移而来的，第四种无法转移是因为已经规定$u$不放装置。那么状态转移方程为 d ( u , 3 ) = s u m { m i n ( d ( v , 0 ) , d ( v , 1 ) , d ( v , 2 ) ) } d(u,3)=sum\{ min( d(v,0),d(v,1),d(v,2) ) \} d(u,3)=sum{min(d(v,0),d(v,1),d(v,2))}

至此本题的代码已经可以写出，部分难点见注释：

#include <bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define pb push_back
#define ins insert
#define lowbit(x) (x&(-x))
#define mkp(x,y) make_pair(x,y)
#define mem(a,x) memset(a,x,sizeof a);
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
typedef pair<int,int> P;
const double eps=1e-8;
const double pi=acos(-1.0);
const int inf=0x3f3f3f3f;
const ll INF=1e18;
const int Mod=1e9+7;
const int maxn=1e4+10;

vector<int> g[maxn];
int d[maxn][4];
int n,c1,c2;

void dfs(int u,int fa){
    d[u][0]=d[u][1]=d[u][3]=0,d[u][2]=c2;  //初始化
    int x=inf;
    for(auto v: g[u]){
        if(v==fa) continue;
        dfs(v,u);  //先递归，回溯过程更新状态
        d[u][0]+=min(d[v][1],d[v][2]);
        int tmp=min(d[v][0],min(d[v][1],d[v][2]));  //先计算出这个下面多次用到
        d[u][1]+=tmp;
        x=min(x,d[v][2]-tmp);
        d[u][2]+=min(tmp,d[v][3]);
        d[u][3]+=tmp;
    }
    d[u][1]+=min(x,c1);  //这样写和直接在上面转移是等价的
}

int main(){
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    while(cin>>n>>c1>>c2 && n){
        for(int i=0;i<=n;i++) g[i].clear();
        for(int i=1,u,v;i<n;i++){
            cin>>u>>v;
            g[u].pb(v);
            g[v].pb(u);
        }
        dfs(1,-1);
        cout<<min(d[1][0],min(d[1][1],d[1][2]))<<endl;
    }
    return 0;
}