ZOJ 3201 Tree of Tree(树形dp + 分组背包)

最新推荐文章于 2020-01-07 20:42:43 发布

玻璃年华Alex

最新推荐文章于 2020-01-07 20:42:43 发布

阅读量739

点赞数

分类专栏： OJ 背包树形dp

本文链接：https://blog.csdn.net/bolininahuaalex/article/details/24037205

版权

OJ 同时被 3 个专栏收录

158 篇文章 0 订阅

订阅专栏

背包

9 篇文章 0 订阅

订阅专栏

树形dp

9 篇文章 0 订阅

订阅专栏

OJ题目：click here ~~

AC_CODE

const int inf = 1<<30;
int n , k;
int x[102];
int dp[102][102];
vector < int > List[102];

void dfs(int u , int father)
{
    int i , j;
    for(i = 0;i < List[u].size();i++)
    {
        int v = List[u][i];
        if(v == father) continue;
        dfs(v , u);
        for(j = k;j >= 1;j--)//一定要倒序
        {
            for(int t = 1;t <= j;t++)
                dp[u][j] = max(dp[u][j] , dp[u][t] + dp[v][j - t]);
        }
    }
}
int main()
{
    while(cin >> n >> k)
    {
        int i , j , a , b;
        for(i = 0;i <= n;i++) List[i].clear();
        memset(dp , 0 , sizeof(dp));
        for(i = 0;i < n;i++)
        {
            scanf("%d",&dp[i][1]);
        }
        for(i = 0;i < n-1;i++)
        {
            scanf("%d%d",&a,&b);
            List[a].push_back(b);
            List[b].push_back(a);
        }
        dfs(0 , -1);
        int ans = -1;
        for(i = 0;i < n;i++)
            ans = max(ans , dp[i][k]);
        cout << ans << endl;
    }
    return 0;
}