快要折磨死我了,还好终于AC了 顿感,这是一道好题
题意:
在一个游戏中,有N 个敌人等着我去杀掉(N≤ 1000)(敌人从0到n-1);杀掉每个敌人所需的时间分别是Ti;这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;
}