hdu1011

大概意思是:有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;
}

猜你喜欢

转载自blog.csdn.net/qq_40570748/article/details/81351807