树上背包2021.5.27

最新推荐文章于 2022-10-28 18:43:03 发布

原创最新推荐文章于 2022-10-28 18:43:03 发布 · 115 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #动态规划

笔记专栏收录该内容

27 篇文章

订阅专栏

本文介绍了一段使用C++编写的代码，通过树形结构和动态规划解决了一个问题，寻找具有特定边权之和的树中最深节点。主要涉及数据结构（树、数组）、算法（动态规划）和函数式编程技巧。

#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;
int dp[3010][3010],sum;
int mo[3010],leaf[3010];
int ne[3010],to[3010],idx,hd[3010],v[3010];
int ans;
void add(int x,int y,int t)
{
    ne[++idx]=hd[x];
    hd[x]=idx;
    v[idx]=t;
    to[idx]=y;
}
void treedp(int top)
{
    if(leaf[top]) sum++;
    int i=hd[top];
    while(i)
    {
        int node=to[i];
        for(int k=0;k<=sum;k++) if(!leaf[node]) dp[node][k]=dp[top][k]-v[i];
                                else dp[node][k+1]=dp[top][k]-v[i]+mo[node];
        treedp(node);
        for(int k=1;k<=sum;k++) dp[top][k]=max(dp[node][k],dp[top][k]);
        i=ne[i];
    }
}
int main()
{
    memset(dp,-64,sizeof dp);
    dp[1][0]=0;
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1,k;i<=n-m;i++)
    {
        scanf("%d",&k);
        for(int j=1,x,y;j<=k;j++)
        {
            scanf("%d%d",&x,&y);
            add(i,x,y);
        }
    }
    for(int i=n-m+1;i<=n;i++)
    {
        scanf("%d",mo+i);
        leaf[i]=1;
    }
    treedp(1);
    for(int i=1;i<=n;i++) if(dp[1][i]>=0) ans=i;
    cout<<ans;
}