dp优化专辑 J -Maximize Game Time [树形dp]

快要折磨死我了，还好终于AC了  顿感，这是一道好题

题意：

在一个游戏中，有N 个敌人等着我去杀掉（N≤ 1000）（敌人从0到n-1）；杀掉每个敌人所需的时间分别是T_i；这N 个敌人之间有父子关系，形成棵树，树的根节点是最终 Boss，一但杀掉他，游戏立即结束；在每一步中，我都可以任选一个敌人杀掉；一但有一个节点的两个儿子被杀掉，我的下一步就不能选择，必须立即杀掉该节点。问，游戏的时间最长可以持续多久。可能是森林~

分析：

如果用

last[v]表示 访问节点v后  的最优解，
hold[v]表示 以v为根的子树在 不访问v  的前提下的最优解
subTree[v]表示  以v为根 的子树的权值和

那么状态转移方程
last[v] = max(last[v1] + subTree[v2] + sum{ hold[v_other] }) {v1、v2、v_other都是v的孩子结点}
hold[v] = max(subTree[v1] + sum{ hold[v_other] })

//AC CODE：

#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<map>

using namespace std;

#define _clr(str,val) memset(str,val,sizeof(str))
const int N=1010;
int t[N],hold[N],last[N],subTree[N],n;
int adj[N][N];
//last[v]表示 最后访问节点v 的最优解
//hold[v]表示 以v为根的子树在不访问v 的前提下的最优解
//subTree[v]表示以v为根的子树的权值和
int father_node;
void _Hold(int v)//hold
{
    int len=adj[v][0],tmp,tt;
    //
    tmp=0;
    for(int j=1; j<=len; j++)
    {
        int u=adj[v][j];
        tmp+=hold[u];
    }
    //
    for(int i=1; i<=len; i++)
    {
        int u=adj[v][i];
        tt=tmp-hold[u];
//        if(hold[v]<last[u]+tt)
//            hold[v]=last[u]+tt;
        if(hold[v]<subTree[u]+tt)
            hold[v]=subTree[u]+tt;

    }
}
void _Last(int v)//last
{
    int len=adj[v][0],tmp,tmp2,tt;
    if(len==0)
        last[v]=subTree[v];
    tmp2=0;
    for(int k=1; k<=len; k++)
    {
        int u_3=adj[v][k];
        tmp2+=hold[u_3];
    }
    for(int i=1; i<=len; i++)
    {
        int u=adj[v][i];
        tmp=0;
        for(int j=1; j<=len; j++)
        {
            if(j==i)
                continue;
            int u_2=adj[v][j];
            tt=tmp2-hold[u]-hold[u_2];
            if(tmp<tt+subTree[u_2])
                tmp=tt+subTree[u_2];
        }
        if(last[v]<last[u]+tmp+t[v])
            last[v]=last[u]+tmp+t[v];
    }
}
void dfs(int v)
{
    int len=adj[v][0];
    for(int i=1; i<=len; i++)
    {
        int u=adj[v][i];
        dfs(u);
    }
    _Hold(v);
    _Last(v);
}
int cal(int v)
{
    subTree[v]=t[v];
    int len=adj[v][0];
    for(int i=1; i<=len; i++)
    {
        int u=adj[v][i];
        subTree[v]+=cal(u);
    }
    return subTree[v];
}
void init()
{
    int f;
    _clr(hold,0);
    _clr(last,0);
    _clr(subTree,0);
    _clr(adj,0);
    for(int i=0; i<n; i++)
    {
        scanf("%d",&t[i]);
    }
    for(int i=0; i<n; i++) //-1父结点
    {
        scanf("%d",&f);
        if(f==-1)
        {
            adj[father_node][0]++;
            adj[father_node][adj[father_node][0]]=i;
        }
        else
        {
            adj[f][0]++;
            adj[f][adj[f][0]]=i;
        }
    }
    //
    for(int i=1; i<=adj[father_node][0]; i++)
    {
        subTree[adj[father_node][i]]=cal(adj[father_node][i]);
    }
    //
}
void print()
{
    //非主树
    int sum=0;
    for(int i=1; i<=adj[father_node][0]; i++)
    {
        if(adj[father_node][i]!=n-1)
            sum+=subTree[adj[father_node][i]];
    }
    //主树
    sum+=last[n-1];
    printf("%d\n",sum);
}
int main()
{
    //freopen("in.txt","r",stdin);
    father_node=1009;
    while(scanf("%d",&n),n)
    {
        init();
        dfs(n-1);
        print();
    }
    return 0;
}