简单树型dp介绍与常用板子+经典例题

最新推荐文章于 2024-02-17 22:55:09 发布
最新推荐文章于 2024-02-17 22:55:09 发布
阅读量237 收藏
点赞数 1
分类专栏： DP 文章标签： 算法 动态规划
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/m0_60921016/article/details/128622222
版权

1

 树的直径的两种求法

看到这个模板想到暑假集训的第一课bwd学长讲的树型dp，当时啥都不会，只是跑去抄了一个两次dfs求树的直径的方法就跑了，现在回想当时bwd学长也讲了树型dp求树的直径只是自己没当回事...

希望自己能好好利用这个寒假~ 队友南京拿了铜牌 棒棒哒 替他们高兴 我也抓紧时间提高一下自己

不然下学期教练都够呛让我继续打了....加油吧

方法一： 两边dfs

这里是利用了图论的简单知识，从树上任意一个点出发离他最远的一个点一定是直径的某个端点

代码如下：这里是这道hdu典例的代码Problem - 2196 (hdu.edu.cn)

#include<iostream>
#include<cstring>
using namespace std;
const int N=2e5+10;
int e[N],ne[N],w[N],h[N],idx;
int n;
int f1[N],f2[N];
int root,len;
void add(int a,int b,int c)
{
  e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
}
void init()
{
  memset(h,-1,sizeof h);
  for(int i=2;i<=n;i++)
  {
    int b,c;
    scanf("%d%d",&b,&c);
    add(i,b,c),add(b,i,c);
  }

}
void dfs(int u,int father,int f[],int x)
{
  f[u]=x;
  for(int i=h[u];~i;i=ne[i])
  {
    int j=e[i];
    if(j==father)continue;
    dfs(j,u,f,x+w[i]);
  }

  if(x>len)
  {
    len=x;
    root=u;
  }
}
void solve()
{
  //找直径的一个端点
   len=-1;
   dfs(1,-1,f1,0);
  //找另一个端点 并更新f1
   len=-1;
   dfs(root,-1,f1,0);
   //更新f2
   len=-1;
   dfs(root,-1,f2,0);

   for(int i=1;i<=n;i++)printf("%d\n",max(f1[i],f2[i]));

}
int main()
{
  while(cin>>n)
  {
    init();
    solve();
  }
}

2.树型dp求法 维护以每个点为根节点的子树种的最长路径和次长路径

1072. 树的最长路径 - AcWing题库

注意法一是不能维护负权值的而法二可以

代码如下：

#include<iostream>
#include<cstring>
using namespace std;
const int N=2e5+10;
int e[N],ne[N],w[N],h[N],idx;
int n;
int f1[N],f2[N];
int root,len;
int res=-0x3f3f3f3f;
void add(int a,int b,int c)
{
  e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
}
void init()
{
  memset(h,-1,sizeof h);
  for(int i=2;i<=n;i++)
  {
    int a,b,c;
    scanf("%d%d%d",&a,&b,&c);
    add(a,b,c),add(b,a,c);
  }

}
void dfs(int u,int father)
{
  for(int i=h[u];~i;i=ne[i])
  {
    int j=e[i];
    if(j==father)continue;
    dfs(j,u);

    if(f1[u]<=f1[j]+w[i])f2[u]=f1[u],f1[u]=f1[j]+w[i];//相等也要换 使得次长也变大
    else if(f2[u]<f1[j]+w[i])f2[u]=f1[j]+w[i];
  }
  res=max(res,f1[u]+f2[u]);

}
void solve()
{
  dfs(1,-1);
  cout<<res;
}
int main()
{
  while(cin>>n)
  {
    init();
    solve();
  }
}

2

树型dp简单模板题2：

树的中心  1073. 树的中心 - AcWing题库

一个节点距离其他点的最远的点 要么是在以自己为根的子树的距离最远点，要么是当前考虑的点向上出发到其父节点，再从父节点出发到最远的点

需要注意的是 由树的拓扑结构可知 某个节点父节点对其贡献是不能计算父节点走”回头路“的距离的，也就是从父节点出发的那条可贡献距离不能再经过该点了

下面是代码：

 

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+10;
int e[N],ne[N],w[N],h[N],idx;
int n;
int f1[N],f2[N],s1[N],s2[N];
int up[N];
void add(int a,int b,int c)
{
  e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
}
void dfs1(int u,int father)
{
   for(int i=h[u];~i;i=ne[i])
   {
    int j=e[i];
    if(j==father)continue;
    dfs1(j,u);
    if(f1[u]<=f1[j]+w[i])
    {
      f2[u]=f1[u],f1[u]=f1[j]+w[i];
      s2[u]=s1[u],s1[u]=j;
    }
    else if(f2[u]<f1[j]+w[i])
    {
      f2[u]=f1[j]+w[i];
      s2[u]=j;
    }
   }

}
void dfs2(int u,int father)
{
  for(int i=h[u];~i;i=ne[i])
  {
    int j=e[i];
    if(j==father)continue;
    if(s1[u]==j)up[j]=max(up[u],f2[u])+w[i];
    else up[j]=w[i]+max(up[u],f1[u]);
    dfs2(j,u);
  }
}
void init()
{
  memset(h,-1,sizeof h);
  for(int i=2;i<=n;i++)
  {
    int a,b,c;
    scanf("%d%d%d",&a,&b,&c);
    add(a,b,c),add(b,a,c);
  }
}

void solve()
{
  dfs1(1,-1);//维护以当前节点为根节点的子树的最长与次长距离
  
 
   
  dfs2(1,-1);//注意和dfs1的不同之处在于从祖宗向孩子更新

  int res=0x3f3f3f3f;
  for(int i=1;i<=n;i++)res=min(res,max(f1[i],up[i]));
  cout<<res;
}
int main()
{
  
    cin>>n;
    init();
    solve();
  
}

了解了基础的模板之后 解水题就比较简单了

下面分享几道树型dp 的简单题

 数字转换 1075. 数字转换 - AcWing题库 
 

简单数论建图+树的直径

#include<bits/stdc++.h>
using namespace std;
const int N=5e5+10,M=1e6+10;
int n;
bool st[N];
int primes[N];
int num;
int h[N],ne[M],w[M],idx,e[N];
int f1[N],f2[N];
int res;
void init()
{
   for(int i=2;i<N;i++)
   {
     if(!st[i])primes[num++]=i;
     for(int j=0;primes[j]<=N/i;j++)
     {
       st[primes[j]*i]=1;
       if(i%primes[j]==0)break;
     }
   }
}
void add(int a,int b,int c)
{
  e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
}
void dfs(int u,int father)
{
  f1[u]=f2[u]=0;
  for(int i=h[u];~i;i=ne[i])
  {
    int j=e[i];
    if(j==father)continue;
    dfs(j,u);
    if(f1[u]<=f1[j]+w[i])f2[u]=f1[u],f1[u]=w[i]+f1[j];
    else if(f2[u]<f1[j]+w[i])f2[u]=f1[j]+w[i];
  }
  res=max(res,f1[u]+f2[u]);
}


int main()
{

  cin>>n;
  init();
  memset(h,-1,sizeof h);
  for(int i=2;i<=n;i++)
  {
    if(!st[i])
    {
      add(i,1,1),add(1,i,1);
      //printf("xixi%d\n",i);
    }
    else{

     int sum=1;
    for(int j=2;j<=i/j;j++)
    {
      if(i%j==0)
      {
         sum+=j;
         if(j!=i/j)sum+=i/j; 
      }
      if(sum>=i)break;
    }

    if(sum<i)
    {
     add(sum,i,1),add(i,sum,1);
      //printf("gaga%d %d\n",i,sum);
    }

    }

  }



  //树的直径
  dfs(1,-1);
  cout<<res;



}

大臣的旅费1207. 大臣的旅费 - AcWing题库 
 

#include<iostream>
#include<cstring>
using namespace std;
const int N=2e6+10;
int dist1[N],dist2[N];
int fa,len;
int n;
int ne[N],w[N],e[N],idx,h[N];
void dfs(int u,int father,int dist[],int x)
{
    dist[u]=x;

    for(int i=h[u];~i;i=ne[i]){
        int j=e[i];
        if(father==j)continue;
        dfs(j,u,dist,x+w[i]);
    }

    if(x>len){
        len=x;
        fa=u;
    }


}
void add(int a,int b,int c)
{
    e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
}
int main()
{
    memset(h,-1,sizeof h);
    cin>>n;
    for(int i=2;i<=n;i++){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c),add(b,a,c);
    }

    len=-1;
    dfs(1,-1,dist1,0);
    len=-1;
    dfs(fa,-1,dist1,0);
    len=-1;
    dfs(fa,-1,dist2,0);


    int res=0;
    for(int i=1;i<=n;i++)res=max(res,max(dist1[i],dist2[i]));

    long long ans=0;
    int st=10;
    for(int i=1;i<=res;i++)
    {
        int tem=st+i;
        ans+=tem;
    }

    cout<<ans;


}

树的直径裸题(蓝桥)

旅游规划1078. 旅游规划 - AcWing题库 
 

判断点在某一条直径上吗? 最大值+次大值==直径长度（树型dp模板2裸题）

#include<bits/stdc++.h>
using namespace std;
const int N=4e5+10;
int e[N],ne[N],w[N],h[N],idx;
int n;
int f1[N],f2[N],s1[N],s2[N];
int up[N];
int res;
void add(int a,int b,int c)
{
  e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
}
void dfs1(int u,int father)
{
   for(int i=h[u];~i;i=ne[i])
   {
    int j=e[i];
    if(j==father)continue;
    dfs1(j,u);
    if(f1[u]<=f1[j]+w[i])
    {
      f2[u]=f1[u],f1[u]=f1[j]+w[i];
      s2[u]=s1[u],s1[u]=j;
    }
    else if(f2[u]<f1[j]+w[i])
    {
      f2[u]=f1[j]+w[i];
      s2[u]=j;
    }
   }
  res=max(res,f1[u]+f2[u]);
}
void dfs2(int u,int father)
{
  for(int i=h[u];~i;i=ne[i])
  {
    int j=e[i];
    if(j==father)continue;
    if(s1[u]==j)up[j]=max(up[u],f2[u])+w[i];
    else up[j]=w[i]+max(up[u],f1[u]);
    dfs2(j,u);
  }
}
void init()
{
  memset(h,-1,sizeof h);
  for(int i=2;i<=n;i++)
  {
    int a,b,c;
    scanf("%d%d",&a,&b);
    add(a,b,1),add(b,a,1);
  }
}

void solve()
{
  dfs1(0,-1);//维护以当前节点为根节点的子树的最长与次长距离



  dfs2(0,-1);//注意和dfs1的不同之处在于从祖宗向孩子更新

  for(int i=0;i<n;i++)//最大值加次大值等于直径
   if(res==(max(f1[i],up[i])+f1[i]+up[i]+f2[i]-max(f1[i],up[i])-min(f1[i],min(up[i],f2[i]))))
    printf("%d\n",i);


}
int main()
{

    cin>>n;
    init();
    solve();

}

没有上司的舞会285. 没有上司的舞会 - AcWing题库

树型dp套简单状态机模型（基础课经典题）

#include<iostream>
#include<cstring>
using namespace std;
const int N=1e5+10;
int f[N][2];
int e[N],ne[N],w[N],idx,h[N];
int root=1;
int n;
int father[N];
void add(int a,int b)
{
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void dfs(int u,int father)
{
    f[u][1]=w[u];
    for(int i=h[u];~i;i=ne[i])
    {
        int j=e[i];
        if(j==father)continue;
        dfs(j,u);
        f[u][0]+=max(f[j][0],f[j][1]);
        f[u][1]+=f[j][0];
    }
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)cin>>w[i];
    memset(h,-1,sizeof h);
    for(int i=2;i<=n;i++)
    {
        int a,b;
        cin>>a>>b;
        add(a,b),add(b,a);
        father[a]=b;
        
    }
    while(father[root])root++;
    
    dfs(root,-1);
    printf("%d",max(f[root][0],f[root][1]));
}

小结：

树型dp个人认为和图论联系紧密一些 常常结合树的直径来考察 比状压好像一点 因为偏向图论的缘故属于dp中码量多一些的

灬德布罗意的猫灬
关注
  • 1
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
【算法归纳】基本算法板子
py今天刷题了吗
04-19 369
文章目录位运算快速幂 位运算 快速幂 LL qmi(LL a,LL b,LL p) { LL res=1 % p; // 边界数据：123456,0,1,因为b=0,所以不会进入while循环， // 若直接定义res=1,p有可能为1，然而任何数%1=0,所以要先模 while(b) { if(b & 1)...
python算法复习(八)----树形dp、位运算
weixin_46253576的博客
05-07 589
# 本章的内容是树形dp，非常重要，熟练掌握后可以解决大部分二叉树问题。 # 以及大数据问题的补充点，以及之前学习过程中没有完成的二进制运算(感谢Golang) class TreeNode(object): def __init__(self, value): self.value = value self.left = None self.right = None def __repr__(self): return s.
树形dp 入门模板题
qq_33969563的博客
05-20 876
树形dp • 树型dp一般先算子树然后进行合并，在实现上与树的后序遍历（这个说法并不准确，因为其实很多都不是二叉树）类似——遍历子树，遍历完之后把子树的值合并给父亲。 • 大部分的树形dp都是利用递归，先确定叶子结点的特征，然后推上去，即状态转移方程一般都是有该点的某些特征 = 其子节点的一些特征 • 有的树形dp以哪个点为根节点无所谓，有的有所谓。 引入 • 给你一棵n个点的树（1号点为根节点），求以点i为根的子树的大小 • f[i]以点i为根的子树的点的个数 • f[i] = 1+Σf[k] (k是i的
迪普科技DPNE培训教材.rar
04-17
迪普科技DPNE全套培训教材
动态规划-树形DP入门-自上而下树形DP
最新发布
estiny_123123的博客
02-17 1287
我们今天所讲的是自顶向下的树形DP，自顶向下的树形动态规划是从树的根节点开始递归计算，通过递归地处理子节点的结果来计算父节点的值。树形动态规划通常使用深度优先搜索（DFS）或广度优先搜索（BFS）遍历树的节点，并结合递归或迭代的方式，通过节点之间的关系逐步计算值。树形动态规划的核心思想是通过递归地处理树的节点，利用节点之间的关系来计算每个节点的值，并最终求解整棵树的问题。如计算树中特定属性的值，例如节点之间的最大距离、特定路径的权重等。他们之间的关系就像一棵以董事长为根的树，父节点就是子节点的直接上司。
树型DP和状态压缩DP+acm.ppt
11-30
树型DP和状态压缩DP+acm.ppt
树型DP（皇宫看守解题报告） 经典
06-13
树型动态规划（Tree Dynamic Programming, 简称树DP）是一种在树结构上进行优化问题求解的算法思想，通常用于解决具有树状结构的数据或问题。在本题中，我们面临的问题是设计一种策略来监控一个由 n 个节点组成的...
经典树型DP状态压缩DP入门
03-16
经典入门 - 树型动态规划和状态压缩动态规划 什么是树型动态规划： 树本身就是一个递归的结构，所以在树上进行动态规划或者递推是在合适不过的事情。 必要条件：子树之间不可以相互干扰，如果本来是相互干扰的，...
动态规划-树型DP经典课件
07-06
动态规划算法中的树型DP
易语言源码树型框+ACCESS例程.rar
02-22
易语言源码树型框+ACCESS例程.rar 易语言源码树型框+ACCESS例程.rar 易语言源码树型框+ACCESS例程.rar 易语言源码树型框+ACCESS例程.rar 易语言源码树型框+ACCESS例程.rar 易语言源码树型框+ACCESS例程.rar
DP-200exam（204题）.docx
02-20
DP-200: Implementing an Azure Data Solution
树形DP学习及例题分析
stldecember的博客
10-27 1149
深度学习树形DP后的一点分析
【夯实算法基础】树形DP入门详解+多道例题剖析
小新的博客
07-27 513
梦想不就是别人眼里不着边际的事吗？它让我变成了现在的自己呀。
动态规划入门（七） 树状DP详解
Hello World!
02-07 543
简而言之是dfs里的动态规划，在深搜过程中动态规划最优解，而树就分为枝叶，由根深搜到枝叶末端。树状dp特点（呈树形发散状）是...
树形dp+树形结构总结
热门推荐
dcx2001的博客
10-18 2万+
总结       最近写了好多树形dp+树形结构的题目，这些题目变化多样能与多种算法结合，但还是有好多规律可以找的。 树形dp一般都是用子树更新父亲（即从下向上更新），那么首先应该考虑的是一个一个子树的更新父亲还是把所有子树都算完了在更新父亲？这就要因题而异了，一般来说有两种情况：1.需要把所有子树的信息都掌握之后再更新子树的就需要把所有子树都算完了在更新父亲。2.而像树上背包这样的问题就需要
树形dp ——树的重心
折腾怪的博客
04-11 827
1.只需要求出最大子树中节点数最小的数目即可题意：有一个国王要把他的领土分给两个儿子，国王的领土是一棵树，N个结点，N-1条边把这些结点连起来，现在大小儿子要选择一个点作为他的首都，那么除首都分别是这两个儿子之外，其他的城市(结点)根据离谁近就归谁所有，如果一样远的话就归大儿子所有，现在假设两个人都采取最优策略，且大儿子先选，问大儿子最多能够得到多少城市？解法：如果大儿子选择了一个点P，那么这个小...
没有上司的舞会——树状DP（Java）
melodyzhan的博客
02-20 420
题目 分析 输入： 5 1 2 3 4 5 1 2 1 3 2 4 2 5 输出： 12 首先建立对应的树 依题意，不能同时选择孩子与双亲 代码 import java.util.Arrays; import java.util.Scanner; public class Main { final static int max = 100010; static class edge{ int to, next; public edge(int to, int next) { //
NE 最新190题库
beiluanyi9456的博客
01-20 154
洛谷 P1122 最大子树和
HJ921004的博客
09-08 247
P1122 最大子树和 题目描述 小明对数学饱有兴趣，并且是个勤奋好学的学生，总是在课后留在教室向老师请教一些问题。一天他早晨骑车去上课，路上见到一个老伯正在修剪花花草草，顿时想到了一个有关修剪花卉的问题。于是当日课后，小明就向老师提出了这个问题： 一株奇怪的花卉，上面共连有NN朵花，共有N-1N−1条枝干将花儿连在一起，并且未修剪时每朵花都不是孤立...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值