#树形DP
文章目录
一、自上而下树形DP
1、模板题
#include<bits/stdc++.h>
using namespace std;
// 例:给一棵有根树,每个点都有一个权值ai,要选择一些点使得权值和最大,且一个点被选择,其儿子不能被选择,问最大值?
// f[i][0],f[i][1] 表示当前点不选和选的方案得到的最大值
const int N = 1e5+2;
int n,a[N];
int f[N][2];
int head[N],cnt;
struct Edge{
int nex,to;
}edge[N<<1];
void add(int from,int to){
edge[++cnt].nex=head[from];
head[from]=cnt;
edge[cnt].to=to;
return;
}
void dfs(int u,int fa){
for(int i=head[u];i;i=edge[i].nex){
int v=edge[i].to;
if(v!=fa)continue;
dfs(v,u);
//代表v这个儿子的状态已经处理好了
f[u][0]+=max(f[v][0],f[v][1]);
//代表u这个点不选,那么v可选可不选
f[u][1]+=f[v][0];
}
}
int main(){
ios::sync_with_stdio(false);cin.tie(nullptr);cout.tie(nullptr);
cin>>n;
for(int i=1;i<=n;i++){
cin>>a[i];
f[i][1]=a[i];
}
for(int i=1;i<n;i++){
int u,v;cin>>u>>v;
add(u,v);add(v,u);
}
dfs(1,0);
cout<<max(f[1][0],f[1][1]);
return 0;
}
2、树形dp关联背包问题
#include<bits/stdc++.h>
using namespace std;
const int N = 1e2+2;
int n,V;
int f[N][N]; // f[u][v]表示以u为根的子树内用了v体积的 所对应的答案
int w[N],v[N]; // w[i]表示第i个节点的权值,v[i]表示第i个节点的体积
int head[N],cnt; // 邻接表存储树的边信息
struct Edge{
int nex,to;
}edge[N<<1];
// 添加边信息到邻接表
void add(int from,int to){
edge[++cnt].nex=head[from];
head[from]=cnt;
edge[cnt].to=to;
return;
}
// 深度优先搜索遍历树
void dfs(int u,int fa){
// 初始化以当前节点u为根的子树内用体积0~V的所有情况的答案为-1
memset(f[u],-1,sizeof(f[u]));
// 如果当前节点的体积小于等于背包容量V,则初始化以当前节点u为根的子树内用当前体积的答案为当前节点的权值
if(v[u]<=V) f[u][v[u]]=w[u];
// 遍历当前节点u的所有子节点
for(int i=head[u];i;i=edge[i].nex){
int v=edge[i].to;
// 如果子节点v等于父节点fa,则跳过,避免重复访问
if(v==fa)continue;
// 递归调用dfs函数,遍历子节点v
dfs(v,u);
// 使用动态规划更新以当前节点u为根的子树内的答案
// 通过遍历当前节点u的子节点的体积v2,更新当前节点u的答案f[u][v1+v2]
vector<int> nf(f[u],f[u]+V+1); // 复制当前答案到nf数组
for(int v1=0;v1<=V;v1++){ // 遍历当前节点u已用体积v1的情况
for(int v2=0;v1+v2<=V;v2++){ // 遍历子节点v用体积v2的情况
nf[v1+v2]=max(nf[v1+v2],f[u][v1]+f[v][v2]); // 更新答案
}
}
// 将更新后的答案nf复制回当前节点u的答案f[u]
for(int v=0;v<=V;v++)f[u][v]=nf[v];
}
return;
}
int main(){
ios::sync_with_stdio(false);cin.tie(nullptr);cout.tie(nullptr);
cin>>n>>V; // 输入树的节点数n和背包容量V
// 输入树的边信息,并构建邻接表
for(int i=1;i<n;i++){
int u,v;cin>>u>>v;
add(u,v);add(v,u);
}
// 从根节点1开始深度优先搜索遍历树
dfs(1,0);
// 计算最终答案,即以根节点为根的子树内的最大权值和
int ans=0;
for(int i=0;i<=V;i++)ans=max(ans,f[1][i]);
// 输出最终答案
cout<<ans;
return 0;
}
二、自下而上树形DP
1、最大独立集
#include<bits/stdc++.h>
using namespace std;
/*
例题;最大独立集
有n名职员,编号为 1~n,他们的关系就像一棵以老板为根的树,父节点就是子节点的直接上司。
每个职员有一个快乐指数,用整数 ai 给出,现在要召开一场周年庆宴会,不过,没有职员愿意和直接上司一起参会。
在满足这个条件的前提下,主办方希望邀请一部分职员参会,使得所有参会职员的快乐指数总和最大,求这个最大值。
即
一个有根树,选择若干个点,使得没有相邻的两个点均被选择。最大化被选择的点的点权和。
状态:
dp[u][0] 表示以u为根节点的子树中选择两两不相邻的若干个点,不选点u的最大权值和
dp[u][1] 表示以u为根节点的子树中选择两两不相邻的若干个点,选了点u的最大权值和
转移:
dp[u][0]=max(dp[v][0],dp[v][1]);
dp[u][1]=a[u]+dp[v][0];
答案:
ans=max(dp[root][0],dp[root][1]);
*/
const int N = 1e5+2;
int n, a[N]; // n表示职员数量,a[i]表示第i个职员的快乐指数
long long dp[N][2]; // dp[u][0]表示以u为根节点的子树中选择两两不相邻的若干个点,不选点u的最大权值和;dp[u][1]表示以u为根节点的子树中选择两两不相邻的若干个点,选了点u的最大权值和
vector<int> e[N]; // 存储树的邻接表
// 深度优先搜索遍历树
void dfs(int u){
// 遍历当前节点u的所有子节点
for(auto v : e[u]){
// 递归调用dfs函数,遍历子节点v
dfs(v);
// 更新dp[u][1]和dp[u][0]
dp[u][1] += dp[v][0]; // 选择了点u,那么点u的快乐指数就会加上所有子节点v未选择点的快乐指数和
dp[u][0] += max(dp[v][0], dp[v][1]); // 不选择点u,那么点u的快乐指数就会加上所有子节点v中选择和不选择点的最大值
}
dp[u][1] += a[u]; // 最后要加上点u的快乐指数
}
int main(){
ios::sync_with_stdio(false);cin.tie(nullptr);cout.tie(nullptr);
cin >> n; // 输入职员数量
set<int> st; // 使用set记录所有职员编号
for(int i = 1; i <= n; i++){
cin >> a[i]; // 输入每个职员的快乐指数
st.insert(i); // 将职员编号加入set中
}
// 构建树的边信息
for(int i = 1, x, y; i < n; i++){
cin >> x >> y;
e[y].push_back(x); // 存储树的邻接表信息
st.erase(x); // 移除已经有父节点的节点,留下根节点
}
int root = *st.begin(); // 找到根节点编号
dfs(root); // 从根节点开始深度优先搜索遍历树
cout << max(dp[root][0], dp[root][1]); // 输出最大快乐指数和
return 0;
}
2、最小点覆盖
#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+3;
int n, dp[N][2]; // dp[u][0]表示以u为根的子树中,u未被覆盖时的最小覆盖数,dp[u][1]表示以u为根的子树中,u被覆盖时的最小覆盖数
vector<int> e[N]; // 存储树的邻接表
// 深度优先搜索遍历树
void dfs(int u, int fa){
// 遍历当前节点u的所有子节点
for(auto v : e[u]){
if(v == fa) continue; // 如果子节点v是当前节点u的父节点,则跳过,避免重复遍历
dfs(v, u); // 递归调用dfs函数,遍历子节点v
// 更新以当前节点u为根的子树中,u未被覆盖和被覆盖时的最小覆盖数
dp[u][0] += dp[v][1]; // 如果当前节点u未被覆盖,则子节点v必须被覆盖,所以dp[u][0]加上dp[v][1]
dp[u][1] += min(dp[v][0], dp[v][1]); // 如果当前节点u被覆盖,则可以选择子节点v被覆盖或不被覆盖中的最小值加上dp[v][1]
}
dp[u][1] += 1; // 当前节点u被覆盖,所以dp[u][1]加上1
}
int main(){
ios::sync_with_stdio(false); cin.tie(nullptr); cout.tie(nullptr);
cin >> n; // 输入节点个数
// 构建树的边信息
for(int i = 1; i < n; i++){
int x, y; cin >> x >> y;
e[x].push_back(y); // 存储树的邻接表信息
e[y].push_back(x); // 存储树的邻接表信息
}
dfs(1, 0); // 从根节点1开始深度优先搜索遍历树
cout << min(dp[1][0], dp[1][1]); // 输出以根节点1为根的子树的最小覆盖数
return 0;
}
3、最小支配集
#include<bits/stdc++.h>
using namespace std;
// dp[u][0]表示以u为根节点的子树中,每一个点都被支配,选了点u的最小点权和
// dp[u][1]表示以u为根节点的子树中,未选点u的最小点权和
// dp[u][2]表示以u为根节点的子树中,除了点u之外的其它点均被支配,但点u没有被支配的最小点权和
using ll = long long;
const int N = 1e5+3;
ll n, dp[N][3], a[N], d[N]; // n表示节点个数,dp数组存储动态规划状态,a数组存储每个节点的点权,d数组存储每个节点的入度
vector<ll> e[N]; // 存储树的邻接表
// 深度优先搜索遍历树
void dfs(int u){
ll minn = 1e18; // 初始最小值设置为一个很大的数
// 遍历当前节点u的所有子节点
for(auto v : e[u]){
dfs(v); // 递归调用dfs函数,遍历子节点v
// 更新以当前节点u为根的子树中,每种情况的最小点权和
dp[u][0] += min({dp[v][0], dp[v][1], dp[v][2]}); // 选了点u的情况下,子树中每个点都被支配的最小点权和是子节点v中各种情况的最小值之和
dp[u][1] += min(dp[v][0], dp[v][1]); // 未选点u的情况下,子树中每个点都被支配的最小点权和是子节点v中选点u或不选点u的最小值之和
minn = min(minn, dp[v][0] - min(dp[v][0], dp[v][1])); // 计算子节点v中除了点u之外的其它点均被支配,但点u没有被支配的最小点权和
dp[u][2] += dp[v][1]; // 子树中除了点u之外的其它点均被支配,但点u没有被支配的最小点权和是子节点v中未选点u的情况下的点权和之和
}
dp[u][0] += a[u]; // 加上当前节点u的点权
dp[u][1] += minn; // 加上当前节点u的最小点权和
}
int main(){
ios::sync_with_stdio(false); cin.tie(nullptr); cout.tie(nullptr);
cin >> n; // 输入节点个数
// 构建树的边信息和节点点权
for(int i = 1; i <= n; i++){
int id, m; cin >> id >> a[id] >> m; // 输入节点编号、节点点权和子节点个数
while(m--){
int x; cin >> x; // 输入子节点编号
e[id].push_back(x); // 存储树的邻接表信息
d[x]++; // 子节点入度加1
}
}
int rt; // 根节点
for(int i = 1; i <= n; i++){
if(d[i] == 0) rt = i; // 找到根节点
}
dfs(rt); // 从根节点开始深度优先搜索遍历树
cout << min(dp[rt][0], dp[rt][1]); // 输出以根节点为根的子树的最小点权和
return 0;
}
三、路径相关树形DP
1、找到n条路经,其权值之和要求最大
#include<bits/stdc++.h>
using namespace std;
/*
给你一个n个点的树。给你m条树上的简单路径,每个路径有个权值ai。
要求选择一些路径,使得每个点至多在一条路径上,并且路径的权值和最大。
状态:
f[i][j]表示以i为节点的子树中,选择了通过节点i的路径j的最大路径权值和。
g[i]表示节点 i 没向 i 上面延伸出去的最优解。
转移:
j=0,i不在任何路径中,f[i][j]=g[v]
j!=0,i不在路径j上,f[i][j]=f[?][j]=g[?]+g[v]
*/
// 定义长整型
using ll = long long;
// 定义树的最大节点数量
const int N = 1e3+2;
// 定义树的邻接表结构体
vector<int> e[N], t;
// 定义操作结构体
struct rig {
vector<int> vec; // 路径节点向量
ll val; // 操作权值
};
// 定义每个节点的操作信息数组
vector<rig> w[N];
// 定义每个节点的最大路径权值和数组
ll dp[N];
// 定义节点数量、操作数量、节点深度数组和父节点数组
int n, m, dep[N] = {1}, f[N];
// 深度优先搜索遍历树
void dfs(int u) {
// 遍历当前节点u的所有子节点
for(auto v : e[u]) {
dfs(v); // 递归调用dfs函数,遍历子节点v
dp[u] += dp[v]; // 更新当前节点u的最大路径权值和,累加子节点v的最大路径权值和
}
// 遍历当前节点u的操作信息
for(auto t : w[u]) {
ll sum = dp[u]; // 初始化sum为当前节点u的最大路径权值和
// 遍历操作涉及的节点路径
for(auto nw : t.vec) {
sum -= dp[nw]; // 减去路径节点的最大路径权值和
// 遍历路径节点的子节点
for(auto v : e[nw]) sum += dp[v]; // 加上路径节点的子节点的最大路径权值和
}
// 更新当前节点u的最大路径权值和,取sum加上操作权值t.val的最大值
dp[u] = max(dp[u], sum + t.val);
}
}
int main() {
// 解除cin与cout的绑定,提高输入输出效率
ios::sync_with_stdio(false);cin.tie(nullptr);cout.tie(nullptr);
// 输入节点数量和操作数量
cin >> n >> m;
// 构建树的结构信息和深度信息
for(int i = 2; i <= n; i++) {
cin >> f[i]; // 输入每个节点的父节点
e[f[i]].push_back(i); // 构建树的邻接表
dep[i] = dep[f[i]] + 1; // 计算每个节点的深度
}
// 读入操作信息
for(int i = 1, x, y; i <= m; i++) {
ll val;
cin >> x >> y >> val; // 输入操作的起点、终点和权值
t.clear(); // 清空临时路径数组
// 计算操作的节点路径
while(x != y) {
if(dep[x] > dep[y]) t.push_back(x), x = f[x];
else t.push_back(y), y = f[y];
}
t.push_back(x); // 加入终点
w[x].push_back({t, val}); // 存储操作信息
}
// 从根节点开始深度优先搜索遍历树
dfs(1);
// 输出根节点的最大路径权值和
cout << dp[1];
return 0;
}
2、找到n条路经,其权值之和要求最小
#include<bits/stdc++.h>
using namespace std;
/*
给你一个n个点的树。给你m条树上的简单路径,每个路径有个权值ai。这里保证每条路径都是从一个点到它的祖先。
要求选择一些路径,使得每个点至多在一条路径上,并且路径的权值和最小。如果不存在输出-1。
状态:
dp[i][j]表示以i为根的子树中全部覆盖,且往根最高延伸到深度j的节点的最小代价。相当于每条路径在最低点考虑。
转移:
不选择以i为最低点的路径:dp[i][j]=dp[v][dep[i]+1]=min(dp[i][j]-dp[v][dep[i]+1]);
对于每一条以i为最低点的路径v,dp[i][dep[v.x]]=min(dp[i][dep[v.x]],v.val+dp[v][dep[i]+1]);
*/
// 定义长整型
using ll = long long;
// 定义树的最大节点数量
const int N = 1e3+2;
// 定义树的邻接表结构体
vector<int> e[N];
// 定义操作结构体
struct rig {
int x; // 路径节点向量
ll val; // 操作权值
};
// 定义每个节点的操作信息数组
vector<rig> w[N];
// 定义每个节点的最大路径权值和数组
ll dp[N][N];
// 定义节点数量、操作数量、节点深度数组和父节点数组
int dep[N],n,m;
// 深度优先搜索遍历树
void dfs(int u) {
// 遍历当前节点u的所有子节点
ll sum = 0; // 记录以当前节点u为根的子树的最小代价
// 初始化每个深度的节点的最小代价为无穷大
for(int i = 1; i <= dep[u]; i++) dp[u][i] = LLONG_MAX;
// 遍历当前节点u的每一个子节点v
for(auto v : e[u]) {
dfs(v); // 递归调用dfs函数,遍历子节点v
sum += dp[v][dep[u]+1]; // 计算以当前节点u为根的子树的最小代价
// 如果子树的最小代价已经足够大,无法继续延伸
if(sum >= LLONG_MAX){
cout << "-1"; // 输出-1表示不存在解
exit(0); // 退出程序
}
// 更新当前节点u的每个深度的节点的最小代价
for(int i = 1; i <= dep[u]; i++) dp[u][i] = min(dp[u][i], dp[v][i] - dp[v][dep[u] + 1]);
}
// 对于每一个操作路径v,更新当前节点u的每个深度的节点的最小代价
for(auto v : w[u]) {
dp[u][dep[v.x]] = min(dp[u][dep[v.x]], sum + v.val); // 更新最小代价
}
// 处理当前节点u的每个深度的节点的最小代价,保证递增性质
for(int i = 2; i <= dep[u]; i++) dp[u][i] = min(dp[u][i], dp[u][i - 1]);
}
int main() {
// 解除cin与cout的绑定,提高输入输出效率
ios::sync_with_stdio(false);cin.tie(nullptr);cout.tie(nullptr);
// 输入节点数量和操作数量
cin >> n >> m;
// 构建树的结构信息和深度信息
for(int i = 2; i <= n; i++) {
int f; // 父节点
cin >> f; // 输入每个节点的父节点
e[f].push_back(i); // 构建树的邻接表
dep[i] = dep[f] + 1; // 计算每个节点的深度
}
// 读入操作信息
for(int i = 1, x, y; i <= m; i++) {
ll val; // 操作权值
cin >> x >> y >> val; // 输入操作的起点、终点和权值
if(dep[x] > dep[y]) swap(x, y); // 交换x和y保证x为较低的节点
w[y].push_back({x, val}); // 存储操作信息
}
// 从根节点开始深度优先搜索遍历树
dfs(1);
// 输出根节点的最小路径权值和
cout << dp[1][1];
return 0;
}
四、换根DP
换根DP,又叫做二次扫描,是树形DP的一种。
特点:以树上的不同点作为根,其解不同,无法通过一次搜索完成答案的求解,因为一次搜索只能得到一个节点的答案。
题目一般为:一棵树…求一哪一个节点为根的时候,xxx最大或最小…
先以树形DP的形式求出以某一个点为根的时候的答案,然后在进行一次自上而下的DFS计算答案。1.树形dp 2.dfs
例题:
给定一棵n个点的无根树,点带权,边带权,求一个点,使得其他点的距离和最小。
距离:a->b的距离为a的点权乘以a->b的路径长度。
分析:
考虑f[i]表示以i为根子树中到i的距离之和。
第一次dfs从下至上递推出各点的f[i]和子树siz[i]
ans[v]=ans[u]-siz[v]*a[i].w+(cnt-siz[v])*a[i].w;
步骤:
1、指定某个节点为根节点;
2、第一次搜索完成预处理(子树大小),同时得到该节点的解;
3、第二次搜索进行换根的dp,由已知解的节点推出相连节点的解。
示例代码:
#include<bits/stdc++.h> // 引入标准 C++ 库的所有头文件
using namespace std;
using ll = long long; // 定义 long long 类型的别名 ll
const int N = 1e3+2; // 定义数组大小常量 N
// 定义边结构体
struct edge{
int nex, to, dis; // 下一条边的索引,目标节点,边的距离
} e[N<<1]; // 边数组
int n, c[N], dist[N]; // 节点数、节点颜色数组、节点到根节点的距离数组
int siz[N], head[N], cnt, tot; // 子树大小数组、邻接表头、计数器、节点颜色总和
// 添加边的函数
void add(int from, int to, int dis){
e[++cnt] = {head[from], to, dis}; // 新建边,更新邻接表
head[from] = cnt; // 更新头节点指针
return;
}
int sum[N]; // 子树节点颜色和数组
ll f[N]; // 子树颜色调整的代价数组
// 深度优先搜索计算子树大小和节点颜色和
void dfs(int x, int fa){
siz[x] = 1; // 初始化节点 x 的子树大小为 1
sum[x] = c[x]; // 初始化节点 x 的子树颜色和为 c[x]
for(int i = head[x]; i; i = e[i].nex){ // 遍历节点 x 的邻接节点
int v = e[i].to; // 邻接节点的索引
if(v == fa) continue; // 如果邻接节点是父节点,则跳过
dist[v] = dist[x] + e[i].dis; // 更新邻接节点到根节点的距离
dfs(v, x); // 递归计算邻接节点的子树大小和颜色和
siz[x] += siz[v]; // 更新节点 x 的子树大小
sum[x] += sum[v]; // 更新节点 x 的子树颜色和
}
return;
}
// 深度优先搜索计算子树颜色调整的代价
void dfs1(int x, int fa){
for(int i = head[x]; i; i = e[i].nex){ // 遍历节点 x 的邻接节点
int v = e[i].to; // 邻接节点的索引
if(v == fa) continue; // 如果邻接节点是父节点,则跳过
f[v] = f[x] - sum[v] * e[i].dis + (tot - sum[v]) * e[i].dis; // 计算邻接节点的代价
}
return;
}
int main(){
ios::sync_with_stdio(false); // 关闭输入输出流同步,加快 I/O 速度
cin.tie(nullptr); // 解除 cin 和 cout 的绑定
cout.tie(nullptr);
cin >> n; // 输入节点数
for(int i = 1; i <= n; i++) cin >> c[i], tot += c[i]; // 输入节点颜色并更新颜色总和
for(int i = 1; i < n; i++){ // 输入 n-1 条边的起点、终点和距离
int a, b, c; cin >> a >> b >> c;
add(a, b, c); // 添加边 (a, b)
add(b, a, c); // 添加边 (b, a)
}
dfs(1, 0); // 计算节点 1 的子树大小和颜色和
ll ans = 1e18; // 初始化答案为一个很大的数
for(int i = 1; i <= n; i++) ans = min(ans, f[i]); // 更新答案为最小的代价
cout << ans; // 输出答案
return 0; // 返回 0,表示程序正常结束
}