大概意思是:有N个洞,M个士兵,每个洞中有a个bug,b个brain,每个士兵可以最多Kill掉20个bug,入口在洞口1处,问可以得到多少个brain。
这些洞是按照图的形式连接的,题中在输入数据的时候给出了图的边。不过根据题意,只有N-1条边,所以刚好是一个树,也就是所谓的树形DP,我是这样理解的,
回到题目,状态方程:value[a][b] = max(value[a][b],value[a][b - k] + value[j][k]),其中a、j之间有边相连。value[a][b]表示以a为根节点的子树,包含b个士兵,所能得到最大brain。 按a来分层进行动态规划。 这里的状态转移方程是根据与之相连的点所决定的,因为只能从当前节点走到与之相连的点。注释在代码中写的很清楚了。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
struct Room
{
int bugs;
int brains;
};
Room rooms[101];
int value[101][101];
int matrix[101][101];
int visited[101];
int n,m;
void dpTree(int u)
{
visited[u]=1;
int r=(rooms[u].bugs+19)/20; //这个r表示当前这个房间里需要留几个士兵,即使是1个虫子也需要一个兵。
for(int i=m;i>=r;i--)
value[u][i]=rooms[u].brains;//只要在这个房子里留够士兵,那么就可以获得全部的brains
for(int v=1;v<=n;v++)
{
if(matrix[u][v] && !visited[v])//选择当前点相邻的点,并且此边没有访问过,注意只能往下走而不能走回去
{
dpTree(v);
for(int j=m;j>=r;j--)
for(int k=1;k<=j-r;k++)
value[u][j]=max(value[u][j],value[u][j-k]+value[v][k]);
}//这个状态转移方程的意思是,第u个房间里留j个士兵所获得brains是由在第u个房间里留j-k个士兵,在与之相连的第v个房间里就k个士兵所获得的brains所决定的。一定记住了,当前的状态仅仅由与其相连的点的状态所决定!!!
}
}
int main()
{
while(scanf("%d %d",&n,&m)!=EOF)
{
if(n==-1)
break;
memset(rooms,0,sizeof(rooms));
memset(matrix,0,sizeof(matrix));
memset(visited,0,sizeof(visited));
memset(value,0,sizeof(value));
for(int i=1;i<=n;i++)
scanf("%d %d",&rooms[i].bugs,&rooms[i].brains);
for(int i=1;i<n;i++)
{
int u,v;
scanf("%d %d",&u,&v);
matrix[u][v]=matrix[v][u]=1;
}
if(m==0)
{
printf("0\n");
continue;
}
dpTree(1);
printf("%d\n",value[1][m]);
}
return 0;
}