wikioi 1163 訪问艺术馆 树形dp

递归建树，由题知该树是一棵二叉树，且除根节点外其它点的度为0或2。

dp[i][j]表示来到第i个走廊（还未走过这条走廊）还剩下j时间，能拿到最大的画的数量。

dp[i][j]=max(dp[i][j],dp[lson[i]][k]+dp[rson][last_time-k])

#include<cstdio>

#include<algorithm>

using namespace std;

int dp[200][700];

int id=0,x,y,s;

struct node

{

    int t,l,r;

    int v;

}tree[200];

void init(int now)

{

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

    tree[now].t=x*2;

    tree[now].v=y;

    if(y!=0) tree[id].l=tree[id].r=-1;

    else

    {

        tree[now].l=++id;init(id);

        tree[now].r=++id;init(id);

    }

}

int dfs(int now,int times)

{

    if(dp[now][times]) return dp[now][times];       //记忆化

    int last=times-tree[now].t;

    if(last<0) return 0;

    if(tree[now].l==-1) return dp[now][times]=min(tree[now].v,last/5);

    int maxn=0;

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

    {

        maxn=max(maxn,dfs(tree[now].l,i)+dfs(tree[now].r,last-i));  /*这里会调用多次已经计算过的值*/

                                                    //计算左儿子i时间，右儿子last-i时间的最大偷画数

    }

    return dp[now][times]=maxn;

}

int main()

{

    scanf("%d",&s);

    init(0);

    printf("%d\n",dfs(0,s));

    return 0;

}

wikioi 1163 訪问艺术馆 树形dp