访问艺术馆（树型动态规划）

访问艺术馆

题目描述：
皮尔是一个出了名的盗画者，他经过数月的精心准备，打算到艺术馆盗画。艺术馆的结构，每条走廊要么分叉为二条走廊，要么通向一个展览室。皮尔知道每个展室里藏画的数量，并且他精确地测量了通过每条走廊的时间，由于经验老道，他拿下一副画需要5秒的时间。你的任务是设计一个程序，计算在警察赶来之前(警察到达时皮尔回到了入口也算)，他最多能偷到多少幅画。

输入描述:
第1行是警察赶到得时间，以s为单位。第2行描述了艺术馆得结构，是一串非负整数，成对地出现：每一对得第一个数是走过一条走廊得时间，第2个数是它末端得藏画数量；如果第2个数是0，那么说明这条走廊分叉为两条另外得走廊。数据按照深度优先得次序给出，请看样例
输出描述:
输出偷到得画得数量
样例输入：
60
7 0 8 0 3 1 14 2 10 0 12 4 6 2
样例输出：
2
数据范围及提示：
s<=600
走廊的数目<=100
思路：
a数组存储点的信息，h数组存储后序遍历的顺序，状态f[i][j]表示以第i个节点（以输入顺序为准）为根，j秒时所能偷到的画的最大数量。
转移时，i枚举点，当a[h[i]]有左右子树时（只需判断有无左子树），用j枚举走过走廊所剩的时间（把走过走廊所用的时间乘2），k枚举分配给左子树的时间进行dp；
当a[h[i]]没有子树时，a[h[i]]为叶节点，用j枚举时间，计算j秒时在该叶节点能偷到画的最大数量（该点的存画数量比较）
最后输出。

#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=610;
int s,tot,n,h[maxn],f[maxn][maxn];
struct node
{
    int t;
    int sum;
    int l,r;
}a[maxn];
int read()
{
    tot++;int t=tot;
    scanf("%d%d",&a[tot].t,&a[tot].sum);
    a[tot].t*=2;
    if(a[tot].sum==0)
    {
        a[t].l=read();
        a[t].r=read();
    }
    return t;
}
void dfs(int x)
{
    if(a[x].l)
    dfs(a[x].l);
    if(a[x].r)
    dfs(a[x].r);
    h[++n]=x;
}
int main()
{
    scanf("%d",&s);
    read();
    dfs(1);
    for(int i=1;i<=n;i++)
    {
        if(a[h[i]].l)
        for(int j=s-a[h[i]].t;j>=0;j--)
          for(int k=0;k<=j;k++)
          f[h[i]][a[h[i]].t+j]=max(f[h[i]][a[h[i]].t+j],f[a[h[i]].l][k]+f[a[h[i]].r][j-k]);
        else
        for(int j=0;j<=s-a[h[i]].t;j++)
        f[h[i]][a[h[i]].t+j]=min(j/5,a[h[i]].sum);
    }
    printf("%d",f[1][s]);
    return 0;
}