P1122 最大子树和题解——树形dp

最新推荐文章于 2024-10-16 00:00:43 发布

domain领域

最新推荐文章于 2024-10-16 00:00:43 发布

阅读量238

点赞数 9

文章标签：算法动态规划

本文链接：https://blog.csdn.net/2401_82728478/article/details/135898234

版权

这次写开始学习树形dp写的一道基础题，下面来详细讲解我的启发。

题目链接：P1122 最大子树和 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

题目大意（摘自洛谷）：一株奇怪的花卉，上面共连有 N 朵花，共有 N−1 条枝干将花儿连在一起，并且未修剪时每朵花都不是孤立的。每朵花都有一个“美丽指数”，该数越大说明这朵花越漂亮，也有“美丽指数”为负数的，说明这朵花看着都让人恶心。所谓“修剪”，意为：去掉其中的一条枝条，这样一株花就成了两株，扔掉其中一株。经过一系列“修剪“之后，还剩下最后一株花（也可能是一朵）。老师的任务就是：通过一系列“修剪”（也可以什么“修剪”都不进行），使剩下的那株（那朵）花卉上所有花朵的“美丽指数”之和最大。

树形dp是基于dfs方式的动态规划，大部分题目的父亲结点的状态由他子结点的状态转移，此题不然，这题题目意思其实就是他的标题要求最大子树和，对于这题，我们可以以这样一种思想：

如上图所示，u的权值和可由v1的最大权值和与v2的最大权值和推过来，v1,v2可由直接相连的子结点推出，特殊的如果是叶子结点他们的最大权值和即为他们本身，所以我们可以推出这样一种状态转移方程： $dp_{u}=\sum max(dp_{v},0)+w_{u}$

其中 $dp_{u}$ 的含义是u结点的最大权值和，而这要与0取最大值表示他可以不选择该结点（如果子结点权值为负数），然后从叶子结点向上递归即可：

#include<set>
#include<map>
#include<stack>
#include<queue>
#include<vector>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
using ll=long long ;
using pii=pair<int,int> ;
#define endl "\n"

constexpr int N=1e5+10;
constexpr int mod=1e9+7;

int n;
ll a[N];
ll dp[N],ans=-1e10;
int h[N],e[N<<1],nex[N<<1],idx;

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

void dfs(int u,int fa)
{
    dp[u]=a[u];
    for(int i=h[u];i!=-1;i=nex[i])
    {
        int j=e[i];
        if(j==fa)continue;
        dfs(j,u);
        
        dp[u]+=max(0LL,dp[j]);
    }
    
    ans=max(ans,dp[u]);
}


void solve()
{
    
    cin>>n;
    
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    
    int t=n-1;
    memset(h,-1,sizeof h);
    while(t--)
    {
        int a,b;
        cin>>a>>b;
        
        add(a,b),add(b,a);
    }
    
    dfs(1,-1);
    cout<<ans<<endl;
}

signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    
    int t;
    t=1;
    
    while(t--)
    solve();
    return 0;
}