POJ1155-树形DP&&背包

技巧：num数组，用来记录在此树上最多选择多少个。其他就是背包了

#include <cstdlib>#include <cstdio>#include <iostream>#include <cstring>using namespace std;const int maxn=3002;int m,n;int len;int dp[maxn][maxn],tem[maxn],a[maxn][maxn],head[maxn],num[maxn];struct node{  int now;   int val;   int next;}tree[9005];void add(int a,int b,int c){  tree[len].now=a;   tree[len].val=b;   tree[len].next=head[c];//建立一个树，链式前进星，    head[c]=len++;}int dfs(int root){   int p;    for(int i=head[root];i!=-1;i=tree[i].next)       {   p=tree[i].now;             dfs(p);           for(int l=0;l<=num[root];l++)                tem[l]=dp[root][l];            for(int k=0;k<=num[root];k++)              for(int j=1;j<=num[p];j++)           {     dp[root][k+j]=max(dp[root][j+k],dp[p][j]+tem[k]-tree[i].val);           }           num[root]+=num[p];       }    return 0;}int main(){int a,b;int k;   while(~scanf("%d%d",&n,&m))   {   len=0;   memset(head,-1,sizeof(head));       for(int i=1;i<=n-m;i++)        {   scanf("%d",&k);            num[i]=0;           for(int c=0;c<k;c++)             {scanf("%d%d",&a,&b);              add(a,b,i);}        }        for(int i=1;i<=n;i++)            for(int j=1;j<=m;j++)            dp[i][j]=-9999999;        for(int i=n-m+1;i<=n;i++)        {scanf("%d",&dp[i][1]);        num[i]=1;}         dfs(1);         //for(int i=1;i<=n-m;i++)            // printf("%d  ",head[i]);         //printf("^^^^\n");         for(int i=m;i>=0;i--)         {    //printf("%d  ",dp[i][1]);             if(dp[1][i]>=0)               {printf("%d\n",i);break;}         }   }    return 0;}