树形DP基础

---

概述

体面：有明显的递归性质和依赖关系
特征：根结点的状态由子节点的状态决定或者，子节点的状态由根节点决定
解决：递归的从根出发，处理完子节点后回溯到他的父亲，这样逐层解决
手段：树上问题必然有递归性质，一般先得知道子节点的相关信息，才能解决父节点

一，树形 X 状态机（基础模型）

题面：选了子节点，父节点就不能选，不选子节点，就无所谓
第一步分析：
1，选和不选：01背包或者01状态机
2，子父的树形关系，直接安排树形DP
单纯求解！！（这里着重分析递归函数）
典例：AcWing 285. 没有上司的舞会
AC 打卡代码

void dfs(int x)
{
    f[x][1]+=happy[x];
    for(int i=h[x];~i;i=nxt[i])
    {
        int s=to[i];
        dfs(s);
        f[x][0]+=max(f[s][0],f[s][1]);
        f[x][1]+=f[s][0];
    }
}

二，树形 X 背包（有依赖的背包问题）

题面：选择某个物品的前提就是选择某个和他人为绑定的物品（约定成树）
分析：选择子节点的前提就是选择父节点
状态表示：$f[root,v]$表示以root为根的子树占用空间为v时的最大价值
传送门：线性DP 之 背包问题

三，树形求解树上问题

引

1，如果输入是无向边，那就正反建两回就好，不过递归传参的时候要知道父节点，别搜回父节点（无限循环警告）
2，树上问题，一定要使用递归

1，树的直径（最长路径）

1，搜索框架以及相关概念：树上问题基础

2，树形DP框架

思路：
1，我们先任取一个点，因为是无根的（有根的就选根就行了）开始做
2，$f[i]$表示从 $i$ 点挂起的最长路径
3，对于搜到的每个点，只有挂起的才是最长的，挂起的直径其实就是向下走的最长路+次长路（dfs一次所有的儿子就能出了）
4，函数要返回单向仅向下的最长路，挂的直径存起就行

$tips：$挂点

int  dfs(int u,int fa)
{
    int d1,d2,d;       
    d1=0;d2=0;                   最大距和次大距
   for(int i=h[u]; ~i ;i=nxt[i])
   {
       int s=to[i];
       if(s==fa)continue;  

       d=dfs(s,u)+w[i];  

       if(d>=d1)d2=d1,d1=d;       d>d1是不对的（d=d1最大值不变，次大值变了）
       else if(d>d2)d2=d;         用个数组记下来也行，不过这道题没这个必要（后面就有了）
   }
    ans=max(ans,d1+d2);
    return d1;
}

2，树的中心（这个用到了父节点信息更新子节点）

题面：树中找到一个点，使得该点到树中其他结点的最远距离最近（好费劲啊~）

1，朴素想法：因为以不同的点为根树是动态的，对每个点求一下最远距离（上文d1），复杂度$\mathcal{O}(n\ast m)$

2，树形DP

• 每一个点在树上其实向上（父节点）和向下（子节点）走

我们用d1[u] , d2[u] , up[u] , p1[u] , p2[u] 分别存一下需要的信息，这些数据存的是：
d1[u]：存下u节点向下走的最长路径的长度
d2[u]：存下u节点向下走的第二长的路径的长度
p1[u]：存下u节点向下走的最长路径是从哪一个节点下去的
p2[u]：存下u节点向下走的第二长的路径是从哪一个节点走下去的（基本上用不到）
up[u]：存下u节点向上走的最长路径的长度

向下走实现：见上文求直径
其实向上走就是求一个点的父节点的不走该节点的最长路径

一个子节点 j 的向上最长路径就是 ：
它的父节点 u 的最长向上路径和最长向下路径取最大值
如果向下最长路径经过了 j 就改为第二长的向下路径
对应代码：

if(p1[u]==j)up[j]=max(up[u],d2[u])+w[i];
else up[j]=max(up[u],d1[u])+w[i];

int dfs(int u,int fa)
{
    for (int i = h[u]; ~i ; i =nxt[i] )
    {
        int s=to[i];
        if(s==fa)continue;
        int d=dfs(s,u)+w[i];
        
        if(d>=d1[u])d2[u]=d1[u],d1[u]=d,p2[u]=p1[u],p1[u]=s;
        else if(d>d2[u])d2[u]=d,p2[u]=s;
    }
    
    return d1[u];
}

int dfs_up(int u,int fa)
{
    for (int i = h[u]; ~i ; i =nxt[i] )
    {
        int s=to[i];
        if(s==fa)continue;
        
        if(p1[u]==s)up[s]=max(up[u],d2[u])+w[i];
        else up[s]=max(d1[u],up[u])+w[i];
        
        dfs_up(s,u);
    }
}

int main()
{
    dfs(1,-1);
    dfs_up(1,-1);.
    ans=1e9;
    for (int i = 1; i <= n; i ++ )ans=min(ans,max(d1[i],up[i]));
    cout << ans;
}