树形DP模型与题目详解acm

最新推荐文章于 2021-08-18 21:04:57 发布
最新推荐文章于 2021-08-18 21:04:57 发布
阅读量287 收藏 1
点赞数
分类专栏： 算法模板 文章标签： 算法 dfs 树上dp acwing c++
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/lafea/article/details/108847432
版权

树形DP问题

#写在前面

##没有上司的舞会

https://www.acwing.com/problem/content/287/
这里用了状态机

----c++版

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
//有两种状态 f[u,0]: 所有从以u为根的子树中选择，并且不选u这个点的方案
//           f[u,1]: 所有从以u为根的子树中选择，并且选u这个点的方案 
//属性：max
//计算：f[u,0]=∑max(f[si,0],f[si,1]);
//      f[u,1]=∑f[si,0];
//o(n)
const int N=6000;
int n;
int happy[N];
int h[N],e[N],ne[N],idx;
int f[N][2];
bool has_father[N];//这题需要自己看一下是否有根节点

void add(int a,int b){
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

void dfs(int u){
    f[u][1]=happy[u];
    for(int i=h[u];i!=-1;i=ne[i]){
        int j=e[i];
        dfs(j);
        
        f[u][0]+=max(f[j][0],f[j][1]);
        f[u][1]+=f[j][0];
    }
}

int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&happy[i]);
    
    memset(h, -1, sizeof h);
    for(int i=0;i<n-1;i++){//n个点的数最大有n-1条边
        int a,b;
        scanf("%d%d",&a,&b);
        has_father[a]=true;
        add(b,a);
    }
    
    int root=1;
    while(has_father[root])root++;
    
    dfs(root);
    
    printf("%d", max(f[root][0],f[root][1]));
    
    return 0;
}

##树的最长路径

https://www.acwing.com/problem/content/1074/

经典找树的直径
在这里插入图片描述
在这里插入图片描述
如果边有负的权值，则需要树形dp
想办法把所有边枚举一遍，选出最大的

在这里插入图片描述
用每个状态dfs表示所有挂在当前点上的路径的最大值？

----c++版

//树的最长路径也可以称为树的直径
#include<iostream>
#include<algorithm>
using namespace std;
#include<cstring>
const int N=10010, M=N*2;
int n;
int h[N], e[M], w[M], ne[M], idx;
int ans;

void add(int a, int b, int c){
    e[idx]=b, w[idx]=c, ne[idx]=h[a], h[a]=idx++;
}

int dfs(int u, int father){
    int dist = 0;//表示从当前点往下走的最大长度
    int d1=0, d2=0;//注意最长距离和次长距离都初始化成0就可以了，如果有负数的边就当它不存在
    //d, d1, d2都是用来求出最大值和次大值
    
    for(int i=h[u]; i!=-1; i=ne[i]){
        int j=e[i];
        if(j==father)continue;//由于是无向图，所以可能往回走
        int d=dfs(j, u)+w[i];
        dist = max(dist, d);// dp 的体现 ？
        
        if(d>=d1)d2=d1,d1=d;
        else if (d>d2)d2=d;
    }
    ans=max(ans, d1+d2);//每一个点的 d1+d2 都可能是答案
    return dist;
}

int main(){
    cin>>n;
    memset(h, -1, sizeof h);
    for(int i=0;i<n-1;i++){
        int a,b,c;
        cin>>a>>b>>c;
        add(a, b, c);
        add(b, a, c);
    }
    
    dfs(1, -1);//dfs不能往上走，不然就死循环了
    
    cout<<ans<<endl;
    return 0;
}

##树的中心

https://www.acwing.com/problem/content/1075/
在这里插入图片描述
向上走就是求一个点的父节点的不走该节点的最长路径，其实我们知道了每一个节点向下走的长度就可以知道向上的最长路径了，一个子节点 j 的向上最长路径就是 它的父节点 u 的最长向上路径和最长向下路径取最大值，如果向下最长路径经过了 j 就改为第二长的向下路径

----c++版

//直观上看起来有点像中心
#include<iostream>
#include<algorithm>
using namespace std;
#include<cstring>
//这题也是暴力枚举，只不过我们用dp的方式让我们枚举得稍微快一些
const int N=10010, M=N*2, inf=0x3f3f3f3f;
int n;
int h[N], e[M], w[M], ne[M], idx;
int d1[N], d2[N], up[N];
int p1[N], p2[N];

void add(int a, int b, int c){
    e[idx]=b, w[idx]=c, ne[idx]=h[a], h[a]=idx++;
}

int dfs_d(int u, int father){
    d1[u]=d2[u]=-inf;
    for(int i=h[u]; i!=-1; i=ne[i]){
        int j=e[i];
        if(j==father)continue;
        int d=dfs_d(j, u)+w[i];
        if(d>d1[u]){
            d2[u]=d1[u]; d1[u]=d;
            p2[u]=p1[u];p1[u]=j;
        }
        else if(d > d2[u]){
            d2[u]=d;
            p2[u]=j;
        }
    }
    if(d1[u]==-inf)d1[u]=d2[u]=0;
    return d1[u];
}

void dfs_u(int u, int father){//每下一个点的父亲的up都是提前算过的，这个是精髓啊
    for(int i=h[u]; i!=-1; i=ne[i]){
        int j=e[i];
        if(j==father)continue;
        if(p1[u]==j)up[j]=max(up[u], d2[u])+w[i];
        else up[j]=max(up[u], d1[u])+w[i];
        
        dfs_u(j, u);
    }
}

int main(){
    cin>>n;
    memset(h, -1, sizeof h);
    for(int i=0; i<n-1; i++){
        int a,b,c;
        cin>>a>>b>>c;
        add(a,b,c);add(b,a,c);
    }
    
    dfs_d(1, -1);
    dfs_u(1, -1);
    
    int res=inf;
    for(int i=1;i<=n;i++)res=min(res, max(d1[i], up[i]));
    
    printf("%d\n", res);
    
    return 0;
}

##数字转换

https://www.acwing.com/problem/content/1077/
在这里插入图片描述

----c++版

#include<iostream>
#include<algorithm>
using namespace std;
const int N=50010;
#include<cstring>
int n;
int h[N], e[N], ne[N], idx;
int sum[N];//存所有数的所有约数之和
int st[N];//因为不只有一棵树，所以需要存一下哪些点是树根
int ans;

void add(int a, int b){
    e[idx]=b; ne[idx]=h[a]; h[a]=idx++;
}

int dfs(int u){
    int d1=0, d2=0;
    for(int i=h[u]; ~i; i=ne[i]){
        int j=e[i];
        int d=dfs(j)+1;
        if(d>=d1)d2=d1, d1=d;
        else if (d>d2)d2=d;
    }
    ans=max(ans, d1+d2);
    return d1;
}

int main(){
    cin>>n;
    for(int i=1; i<=n; i++)
        for(int j=2; j<=n/i; j++)
            sum[i*j]+=i;
    
    memset(h, -1, sizeof h);
    for(int i=2; i<=n; i++)//i需要从2开始枚举，因为不存在0号点，变化的过程中不能有0
        if(i>sum[i]){
            add(sum[i], i);
            st[i]=true;
        }
    
    for(int i=1; i<=n; i++)
        if(!st[i]){//如果是树根
            dfs(i);//变化是单向的，所以是有向边，不用你考虑往上走的问题
        }
    cout<<ans<<endl;
    return 0;
}

##二叉苹果树

https://www.acwing.com/problem/content/1076/

在这里插入图片描述

----c++版

#include<iostream>
#include<algorithm>
using namespace std;
//有依赖背包问题的化简版
#include<cstring>
const int N=110, M=N*2; //虽然有跟节点，但我们不知道边的方向，所以还是建成无相树
int n,m;
int h[N], e[M], w[M], ne[M], idx;
int f[N][N];

void add(int a, int b, int c){
    e[idx]=b, w[idx]=c, ne[idx]=h[a], h[a]=idx++;
}

void dfs(int u, int father){
    for(int i=h[u]; ~i; i=ne[i]){//先物品组
        if(e[i]==father)continue;
        dfs(e[i], u);
        for(int j=m; j>=0; j--)//后体积  必须从大到小
            for(int k=0; k<j; k++)//决策
                f[u][j]=max(f[u][j], f[u][j-k-1]+f[e[i]][k]+w[i]);
    }
}

int main(){
    cin>>n>>m;
    memset(h, -1, sizeof h);
    for(int i=0; i<n-1; i++){
        int a,b,c;
        cin>>a>>b>>c;
        add(a,b,c); add(b, a, c);
    }
    
    dfs(1, -1);
    
    cout<<f[1][m]<<endl;
    return 0;
}

##战略游戏

https://www.acwing.com/problem/content/325/
极大独立集及其相关概念：
https://blog.csdn.net/Richard_for_OI/article/details/79520470

在这里插入图片描述

----c++版

#include<iostream>
#include<algorithm>
using namespace std;
//和没有上司的舞会很像
#include<cstring>
const int N=1510;
int n;
int h[N], e[N], ne[N], idx;
int f[N][2];
bool st[N];

void add(int a, int b){
    e[idx]=b, ne[idx]=h[a], h[a]=idx++;
}

void dfs(int u){
    f[u][0]=0;//每次都要清空
    f[u][1]=1;//有士兵的状态
    for(int i=h[u]; ~i; i=ne[i]){
        int j=e[i];
        dfs(j);
        
        f[u][0]+=f[j][1];
        f[u][1]+=min(f[j][0], f[j][1]);
    }
}

int main(){
    while(scanf("%d", &n)!=-1){
        memset(h, -1, sizeof h);
        idx=0;
        memset(st, 0, sizeof st);
        for(int i=0; i<n; i++){
            int id, cnt;
            scanf("%d:(%d)", &id, &cnt);
            while(cnt--){
                int ver;
                scanf("%d", &ver);
                add(id, ver);
                st[ver]=true;
            }
        }
        int root=0;
        while(st[root])root++;
        dfs(root);
        
        printf("%d\n", min(f[root][0], f[root][1]));
    }
    
    return 0;
}

##皇宫看守

https://www.acwing.com/problem/content/1079/

在这里插入图片描述

----c++版

#include<iostream>
#include<algorithm>
using namespace std;
#include<cstring>
const int N=1510;
int n;
int h[N], e[N], w[N], ne[N], idx;
int f[N][3];
bool st[N];

void add(int a, int b){
    e[idx]=b, ne[idx]=h[a], h[a]=idx++;
}

void dfs(int u){
    f[u][2]=w[u];
    
    for(int i=h[u]; ~i; i=ne[i]){
        int j=e[i];
        dfs(j);
        
        f[u][0]+=min(f[j][1], f[j][2]);
        f[u][2]+=min(min(f[j][0], f[j][1]), f[j][2]);
    }
    
    f[u][1]=1e9;
    for(int i=h[u]; ~i; i=ne[i]){
        int j=e[i];
        f[u][1]=min(f[u][1], f[j][2]+f[u][0]-min(f[j][1], f[j][2]));
    }
}

int main(){
    cin>>n;
    memset(h, -1, sizeof h);
    for(int i=1; i<=n; i++){
        int id, cost, cnt;
        cin>>id>>cost>>cnt;
        w[id]=cost;
        while(cnt--){
            int ver;
            cin>>ver;
            add(id, ver);
            st[ver]=true;//用来找跟节点
        }
    }
    int root=1;
    while(st[root])root++;
    dfs(root);
    cout<<min(f[root][1], f[root][2])<<endl;//根节点没有父节点
    return 0;
}
你哥同学
关注
专栏目录
《英雄编程体验课》第 15 课 | C语言中的数学库
英雄哪里出来
07-05 5万+
C语言中的数学库
动态规划模型总结之树形dp
Hacheylight
01-18 595
刚讲完状压，我们来讲讲树形dp 1.基本概念 树形dpdpdp，顾名思义就是在树上做动态规划，通过若干次的遍历树，得到有用的信息，解决问题 树形dpdpdp的枚举顺序类似线性dpdpdp，也有两个方向： ①：叶→根：根的子节点传递有用的信息给根，由根得出最优解 ②：根→叶：需要取出所有点作为根节点求值，最后根节点获得整棵树的信息 动态规划的顺序一般是后序遍历，即当子节点处理完成再处理父节点 树形d...
【learning】树形dp题目汇总
weixin_30497527的博客
01-17 200
树形dp，就是在树上dp。 解决这类问题的一般步骤 1、确定状态的意思 2、确定状态转移方程 3、确定细节，就是边界和+1-1之类的东西。 这里主要是我树形dp的练习记录。 T1 oiclass1453 二叉苹果树 思路：在dfs的过程中dp。让f[i][j]表示在以i为根的子树内，保留j根树枝，最多保留的苹果树。 状态转移方程：f[i][j]=m...
[ACM] POJ 2342 Anniversary party （树形DP入门题）
逆风的方向 更适合飞翔
10-28 3523
Anniversary party Time Limit: 1000MS   Memory Limit: 65536K Total Submissions: 4410   Accepted: 2496 Description There is going to be a party to celebrate the 80-th Ann
树形DP习题
cheng__yu_的博客
06-25 376
树形DPP1352 没有上司的舞会（树形DP入门题）Computer HDU - 2196 （树形DP经典题） P1352 没有上司的舞会（树形DP入门题） 题意：给定一棵树，每个节点都有自己的权值，父子节点之间只能选择一个，问怎样可以使得整棵树选到的权值最大？ 思路：设 dp[u][0/1]dp[u][0/1]dp[u][0/1]， 0 表示不选择 u 节点，1 表示选择 u 节点。可以这样转移： dp[u][0]=∑max(dp[v][0],dp[v][1])dp[u][0]=\sum max(dp
树形dp专题练习
MekakuCityActors的博客
09-25 378
洛谷p1272 题意：给出一棵含有n个结点的树，求最少删除几条边可以得到一个大小为p的子树 题解：树形dp。dp[u][siz]为得到一个以u为根结点大小为siz的树最少需要删除的边数，那么最后结果就是dp[u][p]的最小值+(u是否等于整棵树的根，如果不是则需要+1用于砍掉与根的联系，否则不需要+1) 洛谷p1273 题意：给出一棵以1为根的含有n个结点的树，树的边权值为非正数，树的叶子结点权...
0x54. 动态规划 - 树形DP（习题详解 × 12）
繁凡さん的博客
08-18 1120
0x54. 动态规划 - 树形DP（习题详解 × 12）
ACM编程必备资料指南：算法、数据结构与STL详解
- **动态规划**：解决最优化问题的核心策略，包括基础DP问题、树形DP、状压DP等。 - **数据结构**：并查集、树状数组、线段树、字典树、Splay树、ST表（Segment Tree）和划分树、树链剖分（Link-Cut Tree）等。 4...
ACM竞赛模板集合：输入输出、算法详解
9. **图论**：包括最小树形图、拓扑排序、欧拉路、曼哈顿路、并查集、割点、桥、双连通分量等概念，这些都是图论中常见的问题模型。 10. **数论**：涉及最大公约数、最小公倍数、高精度计算、幂取模、快速幂、矩阵...
ACM北大训练
09-26
- **定义**: Trie 树是一种用于存储字符串的树形数据结构。 - **应用场景**: 常用于文本搜索、词典等场景。 - **示例题目**: - poj2513: 涉及 Trie 树构建和查找问题。 #### 四、简单搜索 ##### 1. 深度优先搜索 ...
概率dp 树形dp经典题目加解析
04-28
概率dp 树形dp经典题目加解析........高斯求期望...............
树形dp
amunnku8163的博客
07-16 106
我也很久没写树d了 今天切了4题，也就来写下博客 1.树形dp 这是一种在树上的dp，它与线性dp不同，与线性dp的顺序是不同的所以其实树形dp就是 树上dp是一种在树状结构上进行dp的一种，各个阶段呈现树状关系的时候也可以采用树形dp。 2.分类 其实这里也有很多类了，树上背包，删点或者删边类树形DP等等 3.实现 树d的实现其实大多数就是dfs了，对于树的操作也比较普通这里就不在讲了...
树形dp介绍及入门
人工智能方向，计算机视觉
08-15 844
介绍： 树型dp一般先算子树然后进行合并，即：先遍历子树，遍历完之后把子树的值合并给父亲 例一： 给你一棵n个点的树（1号点为根节点），求以点i为根的子树的大小 void dfs(i){ if(i是叶子节点){ f[i]=1;return; } for(k是i的儿子){ dfs(k); f[i]+=f[k]; } f[i]+=1; } f[i]:以点i为根的子树的点的个数 f[i] = 1+Σf[k](k是i的儿子) 例
算法复习——树形dp
banzhangzhi8819的博客
09-18 139
树形dp的状态转移分为两种,一种为从子节点到父节点，一种为父节点到子节点，下面主要讨论子节点到父亲节点的情况： 例题1(战略游戏)： 这是一道典型的由子节点状态转移到父节点的问题，而且兄弟节点之间没有相互影响，我们用f[i][0]/f[i][1]表示i不取/要取时其所在子树总共最少取的节点数,不难得出dp方程： 代码： #include<iostream>...
hdu1054 树形dp
ACM,再见！
02-07 1170
http://acm.hdu.edu.cn/showproblem.php?pid=1054 Problem Description Bob enjoys playing computer games, especially strategic games, but sometimes he cannot find the solution fast enough and then
hdu4340 树形dp
ACM,再见！
02-05 1185
http://acm.hdu.edu.cn/showproblem.php?pid=4340 Problem Description Ant and Bob two army want to capture a country. The country is consist of N cities. To capture the city i, it takes Ant A
算法训练 结点选择 （树形dp）（点权）
LzyRapX
03-08 1657
问题描述 有一棵 n 个节点的树，树上每个节点都有一个正整数权值。如果一个点被选择了，那么在树上和它相邻的点都不能被选择。求选出的点的权值和最大是多少？ 输入格式 第一行包含一个整数 n 。 接下来的一行包含 n 个正整数，第 i 个正整数代表点 i 的权值。 接下来一共 n-1 行，每行描述树上的一条边。 输出格式 输出一个整数，代表选出
树形动态规划（树形DP）入门问题—初探 & 训练
weixin_30814223的博客
04-29 1111
树形DP入门 poj 2342 Anniversary party 先来个题入门一下~ 题意： 某公司要举办一次晚会，但是为了使得晚会的气氛更加活跃，每个参加晚会的人都不希望在晚会中见到他的直接上司，现在已知每个人的活跃指数和上司关系（当然不可能存在环），求邀请哪些人（多少人）来能使得晚会的总活跃指数最大。 解题思路： 任何一个点的取舍可以看作一种决策，那么状态就是在某个...
OI模板大全（普及~省选NOI）
繁凡さん的博客
11-07 1650
hhh刚从某位大佬的洛谷首页偷到了一个好玩的东西 自己对照了一下原来我还有这么多东西没有学呜呜呜我好菜
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值