每日一题 POJ 2486 树形dp 树上背包

最新推荐文章于 2023-08-09 11:22:56 发布

GreyKa

最新推荐文章于 2023-08-09 11:22:56 发布

阅读量186

点赞数

分类专栏：算法竞赛每日一题文章标签：算法数据结构树形DP

本文链接：https://blog.csdn.net/GreyKa/article/details/113803076

版权

算法竞赛每日一题专栏收录该内容

14 篇文章 0 订阅

订阅专栏

每日一题 POJ 2486 树形dp 树上背包

代码并不难写，关键是要想清楚。树形dp的第一个比较完整写完的题，是比较经典的题，来纪念一下。
一开始没想清楚是因为对背包问题的滚动数组优化的理解不够，自然在树上的问题也没思考清楚。
同时因为每个case过后忘记清零，导致re了一发，以后需要注意。
dpr是回到该点，nr是不回到该点，转移方程如代码。

//AC代码 耗时188MS
#include <iostream>
#include<cstring>
#define MAXN 205
using namespace std;
struct EDGE
{
    int to,next;
} edge[MAXN];
int head[MAXN],ptr;
int val[MAXN];
int N,K;
void add_edge(int u,int v)
{
    edge[++ptr].to=v;
    edge[ptr].next=head[u];
    head[u]=ptr;
}
void add(int u,int v)
{
    add_edge(u,v);
    add_edge(v,u);
}
int dpr[105][205],dpnr[105][205];
void dfs(int now,int f)
{
    dpr[now][0]=dpnr[now][0]=val[now];
    for(int p=head[now]; p; p=edge[p].next)
    {
        int to=edge[p].to;
        if(to==f)continue;
        dfs(to,now);
        for(int k=K; k>=1; k--)
        {
            for(int t=2; t<=k; t++)
                dpnr[now][k]=max(dpnr[now][k],dpr[to][t-2]+dpnr[now][k-t]),
                dpr[now][k]=max(dpr[now][k],dpr[to][t-2]+dpr[now][k-t]);
            for(int t=1; t<=k; t++)
                dpnr[now][k]=max(dpnr[now][k],dpnr[to][t-1]+dpr[now][k-t]);
        }
    }
}
void solve()
{
    memset(dpr,0xcf,sizeof(dpr));
    memset(dpnr,0xcf,sizeof(dpnr));
    dfs(1,1);
    int ans=0;
    for(int i=0; i<=K; i++)
        ans=max(ans,dpnr[1][i]);
    cout<<ans<<endl;
    memset(edge,0,sizeof(edge));
    memset(head,0,sizeof(head));
    ptr=0;
}
int main()
{
    while(cin>>N>>K)
    {
        for(int i=1; i<=N; i++)
            cin>>val[i];
        for(int i=1; i<N; i++)
        {
            int a,b; cin>>a>>b;
            add(a,b);
        }
        solve();
    }
}