[HDU] 1561 The more, The Better 树形DP加01分组背包

题目链接:

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

方法:

表面上看输入是一个森林的结构,其实可以用0节点作为父节点将森林中所有的树串起来作为一个树,然后开始DP,设F(P,x)是P为根的子树上选x个能获得的最大值的集合。

      {

      {0}; x==0

F(P,x) =

      {v+P的价值|v属于{F(s,x1)|s是p的直接子节点 && 0<=x1<min( (p为根? x+1:x),m)   }跑一个以一个s作为分组的01分组背包形成的集合};0<x<=min(((p为根? m:m-1)),  p那里最多选出多少)

      }

代码:

#include <iostream>
#include <queue>
#include<algorithm>
using namespace std;
int n,m;
struct Staff
{
    int cost;
    int value;
};
struct Adge
{
    int vetex;
    Adge* nextAdge;
};
struct Node
{
    Adge* firstAdge;
    bool isRoot;
    int value;
    Staff staff[201];
    int StaffCount;
    int maxChose;
};
Node nodes[201];
int myMax(int x,int y)
{
    return x>=y? x:y;
}
void createAdge(int father,int son)
{
    Adge* adge = (Adge*)(malloc(sizeof(Adge)));
    adge->vetex = son;
    if(nodes[father].firstAdge==NULL)
        adge->nextAdge = NULL;
    else
        adge->nextAdge = nodes[father].firstAdge;
    nodes[father].firstAdge = adge;    
}
void DFSDP(int root)
{
    Adge* adge = nodes[root].firstAdge;
    int v,sonCount=0;
    int innerDP[205];
    memset(innerDP,0,sizeof(innerDP));
    while(adge!=NULL)
    {
        v= adge->vetex;
        DFSDP(v);
        nodes[root].maxChose+=nodes[v].maxChose;
        for(int i=m;i>=0;i--)
            for(int j=1;j<=nodes[v].StaffCount;j++)
                if(nodes[v].staff[j].cost<=i)
                    innerDP[i] = myMax(innerDP[i-nodes[v].staff[j].cost]+nodes[v].staff[j].value,innerDP[i]);
        adge=adge->nextAdge;
        sonCount++;
    }
    if(sonCount==0)
    {
        nodes[root].StaffCount++;
        Staff st;
        st.cost=1;
        st.value=nodes[root].value;
        nodes[root].staff[nodes[root].StaffCount]=st;
    }
    else
    {
        int base = root==0?1:0;
        for(int i=0;i<m+base&&i<nodes[root].maxChose+base;i++)
        {
            nodes[root].StaffCount++;
            Staff st;
            st.cost=i+(1-base);
            st.value=innerDP[i]+nodes[root].value*(1-base);
            nodes[root].staff[nodes[root].StaffCount]=st;
        }
    }
}
void main()
{
    Staff st;
    st.cost=st.value = 0;
    while(scanf("%d %d",&n,&m)&&!(n==0 && m==0))
    {
        for(int i=0;i<=200;i++)
        {
            nodes[i].isRoot= false;
            nodes[i].firstAdge=NULL;
            nodes[i].StaffCount=1;
            nodes[i].staff[nodes[i].StaffCount] = st;
            nodes[i].maxChose=1;
        }
        int father,value;
        for(int i=1;i<=n;i++)
        {
            scanf("%d %d",&father,&value);
            nodes[i].isRoot = father==0 ?true :false;
            nodes[i].value = value;
            createAdge(father,i);
        }
        DFSDP(0);
        for(int i=0;i<=nodes[0].StaffCount;i++)
        {
            if(nodes[0].staff[i].cost==m)
            {
                cout<<nodes[0].staff[i].value<<endl;
                break;
            }
        }
    }
}

 感想:实在不知道这个报告应该怎么描述,反正就是知道

原文地址:https://www.cnblogs.com/kbyd/p/3243629.html