hdu1011 数型DP

题目链接

http://acm.hdu.edu.cn/showproblem.php?pid=1011

题目的话大家可以参考其他的博客

我这里仅简单地说一下啊

题意：给出每个房间拥有的BUG数和能得到的能量数，然后给出每个房间的联通图，要到下一个房间必须攻破上一个房间，每个士兵最多消灭20个BUG，就算不足20个BUG也要安排一个士兵

我这里简单地对第一组输入数据做了一个简单地示意图，希望可以帮助你理解

代码的注释写的很详细

#include<stdio.h>
#include<string.h>
#include<vector>
using namespace std;
struct node{
    int v;
    // 记录和本节点的同意父亲节点的相邻的子节点
    int next;
}edge[10010];
//head[i] 记录的是以i为父亲节点的第一个子节点
int head[110];
int dp[110][110];
int n,m,tot;
void add(int s,int t)
{
    edge[tot].v=t;
    edge[tot].next=head[s];
    head[s]=tot++;
}
int max(int a,int b)
{
    return a>b?a:b;
}
int num[110],v[110];
void dfs(int son,int father)
{
    int i,j,tmp,to,k;
    tmp=(num[son]+19)/20;
    for(i=head[son];i!=-1;i=edge[i].next)
    {
        to=edge[i].v;
        if(to==father)
            continue;
        //把当前的节点作为父亲节点接着往下递归
        dfs(to,son);
        for(j=m;j>=tmp;j--)
        {
            for(k=1;j-k>=tmp&&k<=m;k++)
            {
                //DP 背包
                dp[son][j]=max(dp[son][j],dp[son][j-k]+dp[to][k]);
            }
        }
    }
    for(i=tmp;i<=m;i++)
        //记录当前的可以获取的价值
        dp[son][i]+=v[son];
}
int main()
{
    int i,a,b;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        if(n==-1&&m==-1) break;
        memset(head,-1,sizeof(head));
        tot=0;
        for(i=1;i<=n;i++)
            scanf("%d%d",&num[i],&v[i]);
        //建立数的关系
        for(i=1;i<n;i++)
        {
            scanf("%d%d",&a,&b);
            add(a,b);
            add(b,a);
        }
        if(m==0)
        {
            printf("0\n");
            continue;
        }
        memset(dp,0,sizeof(dp));
        // 从根节点找
        dfs(1,0);
        printf("%d\n",dp[1][m]);
    }
    return 0;
}