poj 2486 Apple Tree

http://poj.org/problem?id=2486

树形DP 我一般都是先用邻接表保存树 然后用的都是子树和所有兄弟树的dp方法

自己感觉理解也行呀 但是没见有多少人用呢，也许自己有点另类

老是出错 思维能力还是不够 越来越感觉acm是在练一个人的思维了

题目大意：

给一个树 节点有一定数目的苹果

人在1（根点）最多走K步 问最多可以得到多少苹果

思路：

ans[i][j][0] 表示包括i在内所以其子节点和右边兄弟树 可以走j步 可以存在不回来的情况 的最佳答案

ans[i][j]][1] 这个必须都回来

代码及其注释：

#include<iostream>

#include<stdio.h>

#include<string.h>

#include<queue>

#include<cmath>

#include<stack>

#include<algorithm>

#define LL long long



using namespace std;



const int N=105;

const int MIN=-100000000;

struct node

{

    struct tt *next;

}mem[N];

struct tt

{

    int j;

    struct tt *next;

};

bool had[N];

bool link[N][N];

int applenum[N];

int ans[N][N*2][2];

int a[3][3][3];

void build(int i,int j)

{

    struct tt *t=new tt;

    t->j=j;

    t->next=mem[i].next;

    mem[i].next=t;

}

void Dele(int n)

{

    struct tt *t;

    for(int i=1;i<=n;++i)

    {//cout<<i<<endl;

        while(mem[i].next!=NULL)

        {//cout<<"iii"<<endl;

            t=mem[i].next;

            mem[i].next=t->next;

            delete t;

        }

    }

}

int dp(struct tt *t,int k,int back)

{

   if(t==NULL)//各种 无法走的情况

   return 0;

   if(k<0)

   return 0;

   if(back==1&&k<1)

   return 0;

   if(ans[t->j][k][back]!=-1)

   return ans[t->j][k][back];

   int itemp=0;

   int w=0;

   for(int i=0;i<=k;++i)

   {

       if(i>=1)

       w=applenum[t->j];//要回去时 只有可以回去 才能加上本节点的苹果树

       itemp=max(itemp,w+dp(mem[t->j].next,i-2,1)+dp(t->next,k-i-1,back));//给子树一部分 往兄弟传一部分 注意路径需要的步数减去

       if(back==0)

       {

           w=applenum[t->j];//这是子树不用回来的 其他差不多

           itemp=max(itemp,w+dp(mem[t->j].next,i-1,0)+dp(t->next,k-i-1,1));

       }

   }

   ans[t->j][k][back]=itemp;

   if(t->next!=NULL)

   ans[t->j][k][back]=max(ans[t->j][k][back],dp(t->next,k,back));//如果不走此节点

   return  ans[t->j][k][back];

}

void dfs(int x,int n)

{

    had[x]=true;

    for(int i=1;i<=n;++i)

    {

        if(link[x][i]&&!had[i])

        {

            build(x,i);//建立单向树

            dfs(i,n);

        }

    }

}

int main()

{

    int n,k;

    while(scanf("%d %d",&n,&k)!=EOF)

    {

        for(int i=1;i<=n;++i)

        {

            scanf("%d",&applenum[i]);

        }

        memset(link,false,sizeof(link));

        for(int i=1;i<n;++i)

        {

            int x,y;

            scanf("%d %d",&x,&y);

            link[x][y]=link[y][x]=true;

        }

        if(n==1)

        {

            printf("%d\n",applenum[1]);

            continue;

        }

        memset(had,false,sizeof(had));

        dfs(1,n);

        memset(ans,-1,sizeof(ans));

        printf("%d\n",applenum[1]+dp(mem[1].next,k-1,0));

        Dele(n);

    }

}