之前没做过树状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]);
}
}