JZOJ5484. 【清华集训2017模拟11.26】快乐树

Description

一棵树有n个节点，编号为0到n-1。有一条叫Owaski的狗在树上面走，每一次它可以从一个顶点走到它的任何一个相邻顶点。每个顶点有个可正可负的快乐度，Owaski也有一个快乐度，这个值最开始是0。在他到达一个
顶点的时候，他的快乐度将会加上该顶点的快乐度。当然有时候Owaski的快乐度会是负数，这个时候他会很难受于是会宣泄情绪让快乐度重新变成0。Owaski是一条喜新厌旧的狗，如果它经过了任何一个曾经经过的节
点，它的快乐度不会变化，哪怕这个节点的快乐度为负数。也就是说一个点只有在第一次经过时会对这条狗的快乐度有影响。
Owaski第一个访问的点永远是0号节点（这个点的快乐度也要算），它可以在任何时候离开。它希望它离开时的快乐度尽量高。计算这个快乐度。

Input

第一行一个数n表示点数，点的编号是0到n-1。
接下来一行(n-1)个数，第i个数表示编号为i的点的父亲编号，保证这个编号小于i。
接下来一行n个数，第i个数表示编号为i-1的点的点权。

Output

输出一行一个数表示最大的快乐度。

Sample Input

输入1：
9
0 0 1 1 2 2 5 5
1 2 -3 -7 3 2 7 -1 3
输入2：
20
0 1 1 1 0 3 1 3 4 4 3 6 8 0 12 12 11 7 7
-154011 249645 387572 292156 -798388 560085 -261135 -812756 191481 -165204 81513 -448791 608073 354614 -455750 325999 227225 -696501 904692 -297238

Sample Output

输出1：
17
样例解释1：
最优路径可以是：
0->2->5->8->5->2->6->2->0->1->4

输出2：
3672275

Data Constraint

对于30%的数据，n<=10。
对于60%的数据，n<=20。
对于100%的数据，n<=1000，快乐度均为绝对值不超过1,000,000的整数。

题解

对于一个点，就有两种选择，
要么发泄，要么不发泄，所以设
fx表示在刚到x的时候不发泄，x的子树最大值。
gx表示在刚到x的时候发泄，x的子树最大值。
转移还是比较简单的，
f从f转移过来，g就可以从f，或者g转移过来。

code

#include<queue>#include<cstdio>#include<iostream>#include<algorithm>#include <cstring>#include <string.h>#include <cmath>#include <math.h>#define ll long long#define N 1003#define db double#define P putchar#define G getchar#define mo 998244353using namespace std;char ch;void read(int &n){    n=0;    ch=G();    while((ch<'0' || ch>'9') && ch!='-')ch=G();    ll w=1;    if(ch=='-')w=-1,ch=G();    while('0'<=ch && ch<='9')n=(n<<3)+(n<<1)+ch-'0',ch=G();    n*=w;}int max(int a,int b){return a>b?a:b;}ll min(ll a,ll b){return a<b?a:b;}ll abs(ll x){return x<0?-x:x;}void write(ll x){if(x>9) write(x/10);P(x%10+'0');}void mx(int& x,int y){x=x>y?x:y;}int f[N],g[N],v[N],nxt[N],b[N],n,x;void dfs(int x){    f[x]=v[x];    g[x]=0;    for(int i=b[x];i;i=nxt[i])    {        dfs(i);        f[x]+=max(f[i],0);        g[x]+=max(f[i],g[i]);    }}int main(){    read(n);    for(int i=2;i<=n;i++)        read(x),x++,nxt[i]=b[x],b[x]=i;    for(int i=1;i<=n;i++)        read(v[i]);     dfs(1);    printf("%d",max(f[1],g[1]));}