关闭

【树形dp】PKU-2486-Apple Tree

289人阅读 评论(0) 收藏 举报

在原来很裸的树形dp上加了些限制条件,不过知道状态转移还是很好做的,但状态转移不好想啊,都是看了人家的结题报告后才知道的!

题目

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
#include<map>
#include<stack>
using namespace std;
#define FRE freopen("a.txt","r",stdin);
#define inf 999999999
#define N 105
vector<int> v[N];
int m,dp[2][N][2*N],vis[N];
void dfs(int now)
{
    for(int i=0;i<=m;i++)dp[0][now][i]=dp[1][now][i]=dp[0][now][0];             //在以now为跟的子树里,不管怎样走,其值至少为now的权值
    vis[now]=1;
    for(int i=0;i<v[now].size();i++)
    {
        int u=v[now][i];
        if(vis[u])continue;
        dfs(u);
        for(int j=m-1;j>=0;j--)
        {
            for(int k=m-j;k>=0;k--)
            {
                dp[0][now][j+k+2]=max(dp[0][now][j+k+2],dp[0][now][j]+dp[0][u][k]);             //在以u为跟的子树走了一圈回到now点,然后在除开u的子树再走一圈再回到now
                dp[1][now][j+k+1]=max(dp[1][now][j+k+1],dp[0][now][j]+dp[1][u][k]);             //在除开u的子树走一圈回到now点,然后在以u为跟的子树再走了一圈但不回到now
                dp[1][now][j+k+2]=max(dp[1][now][j+k+2],dp[1][now][j]+dp[0][u][k]);             //在以u为跟的子树走了一圈回到now点,然后在除开u的子树在走一圈但不回到now
            }
        }
    }
}
int main()
{
    //FRE;
    int n;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&dp[0][i][0]);
            v[i].clear();
        }
        for(int i=1;i<n;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            v[x].push_back(y);
            v[y].push_back(x);
        }
        memset(vis,0,sizeof(vis));
        dfs(1);
        printf("%d\n",dp[1][1][m]);
    }
    return 0;
}


0
0

猜你在找
【直播】机器学习&数据挖掘7周实训--韦玮
【套餐】系统集成项目管理工程师顺利通关--徐朋
【直播】3小时掌握Docker最佳实战-徐西宁
【套餐】机器学习系列套餐(算法+实战)--唐宇迪
【直播】计算机视觉原理及实战--屈教授
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之矩阵--黄博士
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之凸优化--马博士
【套餐】Javascript 设计模式实战--曾亮
查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:40101次
    • 积分:774
    • 等级:
    • 排名:千里之外
    • 原创:44篇
    • 转载:0篇
    • 译文:0篇
    • 评论:1条
    文章存档
    最新评论