树形DP
Posted ShadowAA
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了树形DP相关的知识，希望对你有一定的参考价值。
树形DP
树形 DP，即在树上进行的 DP。由于树固有的递归性质，树形 DP 一般都是递归进行的。
例题
没有上司的舞会 洛谷1352
#include<bits/stdc++.h>
using namespace std;
int n,i,x,y,b[6005],f[6005][2];
vector<int>a[6005];
void sc(int x)
for (int i=0;i<a[x].size();i++)
int to=a[x][i];
sc(to);
f[x][0]=f[x][0]+max(f[to][0],f[to][1]);
f[x][1]=f[x][1]+f[to][0];
int main()
cin>>n;
for (i=1;i<=n;i++)
cin>>f[i][1];
for (i=1;i<n;i++)
cin>>x>>y;
a[y].push_back(x);
b[x]=1;
for (i=1;i<=n;i++)
if (!b[i])
break;
sc(i);
cout<<max(f[i][0],f[i][1])<<\'\\n\';
树形dp 入门
今天学了树形dp，发现树形dp就是入门难一些，于是好心的我便立志要发一篇树形dp入门的博客了。
树形dp的概念什么的，相信大家都已经明白，这里就不再多说。直接上例题。
一、常规树形DP
题目描述
某大学有N个职员，编号为1~N。他们之间有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。现在有个周年庆宴会，宴会每邀请来一个职员都会增加一定的快乐指数Ri，但是呢，如果某个职员的上司来参加舞会了，那么这个职员就无论如何也不肯来参加舞会了。所以，请你编程计算，邀请哪些职员可以使快乐指数最大，求最大的快乐指数。
这是一道非常经典的树形dp，首先分析题目，如果上司去了，那么他的所有下司都不能去；如果上司不去，那么他的所有下司去不去无所谓，但要取最大值。接下来就是状态转移方程：
我们设f[i][0]为i不去的情况，f[i][1]为i去的情况，那么
f[i][0]+=max(f[i的下司][1],f[i的下司][0]);
f[i][1]+=f[i的下司]][0];
初始化，如果i去，那么显然f[i][1]=i的快乐指数，如果i不去那么f[i][1]=0；
最后贴上代码
1 #include<iostream> 2 #include<cstring> 3 #include<cmath> 4 #include<string> 5 #include<cstdio> 6 using namespace std; 7 struct edge 8 { 9 int num,child[6001]; 10 }g[6001]; 11 int n; 12 int a[10000]; 13 int tree[10000]; 14 int f[6001][2]; 15 int aa,bb,root; 16 void dp(int t) 17 { 18 f[t][1]=a[t];//初始化 19 f[t][0]=0;//初始化 20 for(int i=1;i<=g[t].num;i++) 21 { 22 dp(g[t].child[i]); 23 f[t][0]+=max(f[g[t].child[i]][1],f[g[t].child[i]][0]);//状态转移 24 f[t][1]+=f[g[t].child[i]][0];//状态转移 25 } 26 } 27 int main() 28 { 29 cin>>n; 30 for(int i=1;i<=n;i++) 31 { 32 scanf("%d",&a[i]); 33 } 34 while(scanf("%d%d",&aa,&bb)) 35 { 36 if(aa==0&&bb==0) break; 37 g[bb].num++;//记录孩子结点的个数 38 g[bb].child[g[bb].num]=aa;//储存孩子结点 39 tree[aa]=bb;//记录父亲结点 40 } 41 root=1; 42 while(tree[root]) root++;//找出树根 43 dp(root);//从树根开始dp 44 int ans=max(f[root][1],f[root][0]); 45 cout<<ans; 46 return 0; 47 }
题目描述
Bob喜欢玩电脑游戏，特别是战略游戏。但是他经常无法找到快速玩过游戏的办法。现在他有个问题。
他要建立一个古城堡，城堡中的路形成一棵树。他要在这棵树的结点上放置最少数目的士兵，使得这些士兵能了望到所有的路。
注意，某个士兵在一个结点上时，与该结点相连的所有边将都可以被了望到。
请你编一程序，给定一树，帮Bob计算出他需要放置最少的士兵.
和上个题一模一样，就是初始化不同，上个题每个人都有不同的权值，而这个题每个人的权值都为1；
状态转移一模一样 ；初始化，f[i][1]=1，f[i][1]=0；
最后上代码
1 #include<iostream> 2 #include<cstdio> 3 #include<cmath> 4 #include<string> 5 #include<cstring> 6 using namespace std; 7 struct edge 8 { 9 int num,child[1501]; 10 }g[1501]; 11 int a[1501]; 12 int f[1501][2]; 13 int x,y; 14 int n; 15 void dp(int t) 16 { 17 f[t][1]=1;//初始化为1 18 f[t][0]=0; 19 for(int i=1;i<=g[t].num;i++) 20 { 21 dp(g[t].child[i]); 22 f[t][0]+=f[g[t].child[i]][1];//状态转移 23 f[t][1]+=min(f[g[t].child[i]][1],f[g[t].child[i]][0]);//状态转移 24 } 25 } 26 int main() 27 { 28 cin>>n; 29 for(int i=1;i<=n;i++) 30 { 31 scanf("%d",&x); 32 scanf("%d",&g[x].num);//记录该结点的孩子个数 33 for(int j=1;j<=g[x].num;j++) 34 { 35 scanf("%d",&g[x].child[j]);//存储该结点的孩子 36 a[j]=i;//记录父亲 37 } 38 } 39 int root=0; 40 while(a[root]) root++;//找出根节点 41 dp(root); 42 int ans=min(f[root][1],f[root][0]); 43 cout<<ans; 44 return 0; 45 }
洛谷P2458 [SDOI2006]保安站岗
题目描述
五一来临，某地下超市为了便于疏通和指挥密集的人员和车辆，以免造成超市内的混乱和拥挤，准备临时从外单位调用部分保安来维持交通秩序。
已知整个地下超市的所有通道呈一棵树的形状；某些通道之间可以互相望见。总经理要求所有通道的每个端点（树的顶点）都要有人全天候看守，在不同的通道端点安排保安所需的费用不同。
一个保安一旦站在某个通道的其中一个端点，那么他除了能看守住他所站的那个端点，也能看到这个通道的另一个端点，所以一个保安可能同时能看守住多个端点（树的结点），因此没有必要在每个通道的端点都安排保安。
编程任务：
请你帮助超市经理策划安排，在能看守全部通道端点的前提下，使得花费的经费最少。
难度提升，如果看不懂可以先放一放
分析：
我们发现，要使所有点最终全部被覆盖，那么无非有3种状态：
（以下全部都是对于要覆盖任意一个以x为根的子树来说的）
（其中我们设y节点为y的儿子，fa为x的父亲）
1.x节点被自己覆盖，即选择x点来覆盖x点
2.x节点被儿子y覆盖，即选择y点来覆盖x点
3.x节点被父亲fa覆盖，即选择fa点来覆盖x点借此三种状态，我们可以设f[x][0/1/2]为让以x为根的子树中的节点全部被覆盖，且x点的被覆盖情况为1/2/3时的最小代价
为了方便，我们不妨设这三种情况分别为：
1.f[x][0]---对应上面的1
2.f[x][1]---对应上面的2
3.f[x][2]---对应上面的3
既然是DP，总是有转移方程的，我们想一下dp方程要如何设计
设计状态转移方程：
(1):对应上面的1.
f[x][0]=∑ min(f[y][0],f[y][1],f[y][2]) + val[x]
其中val[x]是选择x点的代价
我们很容易想到，在节点x被选择之后，我们就可以无拘无束了（蛤?），也就是说对于x儿子节点y的状态可以不去考虑，因为x节点被选择之后y节点无论如何也会被覆盖到，所以我们在儿子y的所有状态里取min，累加起来就行了
(2):对应上面的3（先讲3，因为2比较难以理解，放到了后面）
f[x][2]=∑ min(f[y][0],f[y][1])
为什么3情况对应的转移方程要这样写呢?
我们不妨这样理解，对于x节点我们让它的父亲节点fa覆盖它，那么根据我们的状态设计，此时必须要满足以x的儿子y为根的子树之中所有点已经被覆盖
那么这时就转化为一个子问题，要让y子树满足条件，只有两种决策：要么y被y的儿子覆盖，要么被y自己覆盖（即选择y节点），只需要在y的这两种状态取min累加就可以了
(3):对应上面的2（DuangDuangDuang 敲黑板划重点啦）
f[x][1]=∑ min(f[y][0],f[y][1])，如果选择的全部都是f[y][1],要再加上min(f[y][0]-f[y][1])
到了这里，我们就要回顾一下我们设计的dp状态了：
设f[x][0/1/2]为让以x为根的子树中的节点全部被覆盖，且x点的被覆盖情况为1/2/3时的最小代价
先提示一下，如果你理解了下面，那么本题是很简单的。。如果你没理解，就返回到这里再看一遍吧，我就在这里等着你
对于此时的状态，f[x][1]代表对于节点x让x被自己的儿子覆盖，那么和分析（2）一样，都要先满足此时以y的子树已经满足了条件，才能进行转移，这就是前面那部分：∑ min(f[y][0],f[y][1])的来历，那么后面那一长串又是怎么回事呢?
我们可以这样理解，此时既然要保证x点是被自己的儿子覆盖的，那么如果此时y子树已经满足了全部被覆盖，但是y此时被覆盖的状态却是通过y节点自己的儿子达到的，那么x就没有被儿子y覆盖到，那么我们不妨推广一下，如果x所有的儿子y所做的决策都不是通过选择y点来满足条件，那么我们就必须要选择x的一个子节点y，其中y满足f[y][0]-f[y][1]最小，并把这个最小的差值累加到f[x][1]中去，这样才能使得x点被自己的儿子覆盖，状态f[x][1]也才能合理地得到转移
好了，如果你还是没有太懂这个（3）的设计过程，请你回到之前再仔细看几遍
如果你已经理解了上面，那么恭喜你这个题，你已经A掉了
因为转移方程既然有了，那么我们就只需要最后的答案了
由于题目中没有说这棵树的根节点是哪个，所以你可以默认1就是根，或者开一个数组在加边的时候记录一下每个点的入度，最后没有入度的点就是根（但好像没有区别，毕竟我A掉了）
最后答案即为min(f[root][0],f[root][1])
1 #include<iostream> 2 #include<cstdio> 3 #include<cmath> 4 #include<string> 5 #include<cstring> 6 using namespace std; 7 struct edge 8 { 9 int num,child[6001]; 10 }g[6001]; 11 int a[6001]; 12 int dp[6001][3]; 13 int root[6001]; 14 int n,aa; 15 void dfs(int x) 16 { 17 dp[x][0]=a[x]; 18 int cc=1000000000; 19 int num=0; 20 for(int i=1;i<=g[x].num;i++) 21 { 22 dfs(g[x].child[i]); 23 int t=min(dp[g[x].child[i]][0],dp[g[x].child[i]][1]); 24 dp[x][0]+=min(t,dp[g[x].child[i]][2]); 25 dp[x][1]+=t; 26 dp[x][2]+=t; 27 if(dp[g[x].child[i]][0]<dp[g[x].child[i]][1]) num++; 28 else cc=min(cc,dp[g[x].child[i]][0]-dp[g[x].child[i]][1]); 29 } 30 if(num==0) dp[x][1]+=cc; 31 } 32 int main() 33 { 34 scanf("%d",&n); 35 for(int i=1;i<=n;i++) 36 { 37 scanf("%d",&aa); 38 scanf("%d",&a[aa]); 39 scanf("%d",&g[aa].num); 40 for(int j=1;j<=g[aa].num;j++) 41 { 42 scanf("%d",&g[aa].child[j]); 43 root[g[aa].child[j]]=aa; 44 } 45 } 46 aa=1; 47 while(root[aa]) aa++; 48 dfs(aa); 49 cout<<min(dp[aa][0],dp[aa][1]); 50 return 0; 51 }
二、树形背包问题(在树上进行分组背包处理)
未完待续~~~
以上是关于树形DP的主要内容，如果未能解决你的问题，请参考以下文章