冲刺阶段:补漏(树状dp)

之前没做过树状dp,虽然做了很多题目,现在刚碰到第一个。
地址: 点击打开链接
这个题目其实很明显的在暗示树状dp了,很入门,但是需要思考的思考的东西还是很多,比如这个是有限背包问题,因此在更新状态的时候必须要倒着更新。dp[father][j] = max(dp[father][j],dp[father][j-s]+dp[son][s]) 当前节点分配的最大值等于 max(现状,满足当前节点要求的情况下,把剩余的士兵分配给子节点的结果)

代码:

#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
#define MAX 155
int N,M;
struct Node{
    int needPerson ;
    int value ;

}nodes[MAX];
int vis[MAX];
int dp[MAX][MAX];
vector<int> Map[MAX];

void dfs(int root){
    for(int i = nodes[root].needPerson ; i <=M ; i++)
    {
        dp[root][i] = nodes[root].value;
    }
    vis[root] = 1 ;
    for(int i = 0 ; i <Map[root].size()  ; i ++)
    {
        int son = Map[root][i];
        if(vis[son])
            continue;
        dfs(son);
        for(int j = M ; j>=nodes[root].needPerson ; j-- )//这个地方必须倒着,有限背包
        {
            for(int k = j - nodes[root].needPerson; k>= 1 ; k --)//这个地方怎么样都可以,但是k不能取0。这个和下面那个地方一个意思,我没看到这个限制,就是M=0,那么必然为0.
            {
                    dp[root][j] = max(dp[root][j],dp[root][j-k]+dp[son][k]);
            }
        }
    }


}
int main(){
    while(~scanf("%d%d",&N,&M)&&N!=-1&&M!=-1)
    {
        int bugs,a,b;
        for(int i = 1 ; i <= N ;i ++ )
        {
            scanf("%d%d",&bugs,&nodes[i].value);
            nodes[i].needPerson = (bugs+19)/20;

        }
        for(int i = 0 ; i <= N ; i ++)
        {
            Map[i].clear();
        }
        for(int i = 0 ; i <N-1 ; i ++)
        {
                scanf("%d%d",&a,&b);
                Map[a].push_back(b);
                Map[b].push_back(a);
        }
        if(M == 0 )//不能取到0,感觉这个题目不严谨,因为如果没有虫子的话,M为0,那么肯定可以拿到宝藏,嗯
        {
            printf("0\n");
            continue ;
        }
        memset(vis,0,sizeof(vis));
        memset(dp,0,sizeof(dp));
        dfs(1);

        printf("%d\n",dp[1][M]);
    }

}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值